backbone/stem/conv1/weight: normal(0, 1.4142135623730951 / fan_out), shape=[7, 7, 3, 64], axes=FanAxes(in_axis=-2, out_axis=-1, batch_axis=())
backbone/stem/norm1/scale: constant(1.0)
backbone/stem/norm1/bias: constant(0.0)
backbone/stem/norm1/moving_mean: constant(0.0)
backbone/stem/norm1/moving_variance: constant(1.0)
backbone/stage0/block0/conv1/weight: normal(0, 1.4142135623730951 / fan_out), shape=[3, 3, 64, 64], axes=FanAxes(in_axis=-2, out_axis=-1, batch_axis=())
backbone/stage0/block0/norm1/scale: constant(1.0)
backbone/stage0/block0/norm1/bias: constant(0.0)
backbone/stage0/block0/norm1/moving_mean: constant(0.0)
backbone/stage0/block0/norm1/moving_variance: constant(1.0)
backbone/stage0/block0/conv2/weight: normal(0, 1.4142135623730951 / fan_out), shape=[3, 3, 64, 64], axes=FanAxes(in_axis=-2, out_axis=-1, batch_axis=())
backbone/stage0/block0/norm2/scale: constant(1.0)
backbone/stage0/block0/norm2/bias: constant(0.0)
backbone/stage0/block0/norm2/moving_mean: constant(0.0)
backbone/stage0/block0/norm2/moving_variance: constant(1.0)
backbone/stage0/block1/conv1/weight: normal(0, 1.4142135623730951 / fan_out), shape=[3, 3, 64, 64], axes=FanAxes(in_axis=-2, out_axis=-1, batch_axis=())
backbone/stage0/block1/norm1/scale: constant(1.0)
backbone/stage0/block1/norm1/bias: constant(0.0)
backbone/stage0/block1/norm1/moving_mean: constant(0.0)
backbone/stage0/block1/norm1/moving_variance: constant(1.0)
backbone/stage0/block1/conv2/weight: normal(0, 1.4142135623730951 / fan_out), shape=[3, 3, 64, 64], axes=FanAxes(in_axis=-2, out_axis=-1, batch_axis=())
backbone/stage0/block1/norm2/scale: constant(1.0)
backbone/stage0/block1/norm2/bias: constant(0.0)
backbone/stage0/block1/norm2/moving_mean: constant(0.0)
backbone/stage0/block1/norm2/moving_variance: constant(1.0)
backbone/stage1/block0/conv1/weight: normal(0, 1.4142135623730951 / fan_out), shape=[3, 3, 64, 128], axes=FanAxes(in_axis=-2, out_axis=-1, batch_axis=())
backbone/stage1/block0/norm1/scale: constant(1.0)
backbone/stage1/block0/norm1/bias: constant(0.0)
backbone/stage1/block0/norm1/moving_mean: constant(0.0)
backbone/stage1/block0/norm1/moving_variance: constant(1.0)
backbone/stage1/block0/conv2/weight: normal(0, 1.4142135623730951 / fan_out), shape=[3, 3, 128, 128], axes=FanAxes(in_axis=-2, out_axis=-1, batch_axis=())
backbone/stage1/block0/norm2/scale: constant(1.0)
backbone/stage1/block0/norm2/bias: constant(0.0)
backbone/stage1/block0/norm2/moving_mean: constant(0.0)
backbone/stage1/block0/norm2/moving_variance: constant(1.0)
backbone/stage1/block0/downsample/conv/weight: normal(0, 1.4142135623730951 / fan_out), shape=[1, 1, 64, 128], axes=FanAxes(in_axis=-2, out_axis=-1, batch_axis=())
backbone/stage1/block0/downsample/norm/scale: constant(1.0)
backbone/stage1/block0/downsample/norm/bias: constant(0.0)
backbone/stage1/block0/downsample/norm/moving_mean: constant(0.0)
backbone/stage1/block0/downsample/norm/moving_variance: constant(1.0)
backbone/stage1/block1/conv1/weight: normal(0, 1.4142135623730951 / fan_out), shape=[3, 3, 128, 128], axes=FanAxes(in_axis=-2, out_axis=-1, batch_axis=())
backbone/stage1/block1/norm1/scale: constant(1.0)
backbone/stage1/block1/norm1/bias: constant(0.0)
backbone/stage1/block1/norm1/moving_mean: constant(0.0)
backbone/stage1/block1/norm1/moving_variance: constant(1.0)
backbone/stage1/block1/conv2/weight: normal(0, 1.4142135623730951 / fan_out), shape=[3, 3, 128, 128], axes=FanAxes(in_axis=-2, out_axis=-1, batch_axis=())
backbone/stage1/block1/norm2/scale: constant(1.0)
backbone/stage1/block1/norm2/bias: constant(0.0)
backbone/stage1/block1/norm2/moving_mean: constant(0.0)
backbone/stage1/block1/norm2/moving_variance: constant(1.0)
backbone/stage2/block0/conv1/weight: normal(0, 1.4142135623730951 / fan_out), shape=[3, 3, 128, 256], axes=FanAxes(in_axis=-2, out_axis=-1, batch_axis=())
backbone/stage2/block0/norm1/scale: constant(1.0)
backbone/stage2/block0/norm1/bias: constant(0.0)
backbone/stage2/block0/norm1/moving_mean: constant(0.0)
backbone/stage2/block0/norm1/moving_variance: constant(1.0)
backbone/stage2/block0/conv2/weight: normal(0, 1.4142135623730951 / fan_out), shape=[3, 3, 256, 256], axes=FanAxes(in_axis=-2, out_axis=-1, batch_axis=())
backbone/stage2/block0/norm2/scale: constant(1.0)
backbone/stage2/block0/norm2/bias: constant(0.0)
backbone/stage2/block0/norm2/moving_mean: constant(0.0)
backbone/stage2/block0/norm2/moving_variance: constant(1.0)
backbone/stage2/block0/downsample/conv/weight: normal(0, 1.4142135623730951 / fan_out), shape=[1, 1, 128, 256], axes=FanAxes(in_axis=-2, out_axis=-1, batch_axis=())
backbone/stage2/block0/downsample/norm/scale: constant(1.0)
backbone/stage2/block0/downsample/norm/bias: constant(0.0)
backbone/stage2/block0/downsample/norm/moving_mean: constant(0.0)
backbone/stage2/block0/downsample/norm/moving_variance: constant(1.0)
backbone/stage2/block1/conv1/weight: normal(0, 1.4142135623730951 / fan_out), shape=[3, 3, 256, 256], axes=FanAxes(in_axis=-2, out_axis=-1, batch_axis=())
backbone/stage2/block1/norm1/scale: constant(1.0)
backbone/stage2/block1/norm1/bias: constant(0.0)
backbone/stage2/block1/norm1/moving_mean: constant(0.0)
backbone/stage2/block1/norm1/moving_variance: constant(1.0)
backbone/stage2/block1/conv2/weight: normal(0, 1.4142135623730951 / fan_out), shape=[3, 3, 256, 256], axes=FanAxes(in_axis=-2, out_axis=-1, batch_axis=())
backbone/stage2/block1/norm2/scale: constant(1.0)
backbone/stage2/block1/norm2/bias: constant(0.0)
backbone/stage2/block1/norm2/moving_mean: constant(0.0)
backbone/stage2/block1/norm2/moving_variance: constant(1.0)
backbone/stage3/block0/conv1/weight: normal(0, 1.4142135623730951 / fan_out), shape=[3, 3, 256, 512], axes=FanAxes(in_axis=-2, out_axis=-1, batch_axis=())
backbone/stage3/block0/norm1/scale: constant(1.0)
backbone/stage3/block0/norm1/bias: constant(0.0)
backbone/stage3/block0/norm1/moving_mean: constant(0.0)
backbone/stage3/block0/norm1/moving_variance: constant(1.0)
backbone/stage3/block0/conv2/weight: normal(0, 1.4142135623730951 / fan_out), shape=[3, 3, 512, 512], axes=FanAxes(in_axis=-2, out_axis=-1, batch_axis=())
backbone/stage3/block0/norm2/scale: constant(1.0)
backbone/stage3/block0/norm2/bias: constant(0.0)
backbone/stage3/block0/norm2/moving_mean: constant(0.0)
backbone/stage3/block0/norm2/moving_variance: constant(1.0)
backbone/stage3/block0/downsample/conv/weight: normal(0, 1.4142135623730951 / fan_out), shape=[1, 1, 256, 512], axes=FanAxes(in_axis=-2, out_axis=-1, batch_axis=())
backbone/stage3/block0/downsample/norm/scale: constant(1.0)
backbone/stage3/block0/downsample/norm/bias: constant(0.0)
backbone/stage3/block0/downsample/norm/moving_mean: constant(0.0)
backbone/stage3/block0/downsample/norm/moving_variance: constant(1.0)
backbone/stage3/block1/conv1/weight: normal(0, 1.4142135623730951 / fan_out), shape=[3, 3, 512, 512], axes=FanAxes(in_axis=-2, out_axis=-1, batch_axis=())
backbone/stage3/block1/norm1/scale: constant(1.0)
backbone/stage3/block1/norm1/bias: constant(0.0)
backbone/stage3/block1/norm1/moving_mean: constant(0.0)
backbone/stage3/block1/norm1/moving_variance: constant(1.0)
backbone/stage3/block1/conv2/weight: normal(0, 1.4142135623730951 / fan_out), shape=[3, 3, 512, 512], axes=FanAxes(in_axis=-2, out_axis=-1, batch_axis=())
backbone/stage3/block1/norm2/scale: constant(1.0)
backbone/stage3/block1/norm2/bias: constant(0.0)
backbone/stage3/block1/norm2/moving_mean: constant(0.0)
backbone/stage3/block1/norm2/moving_variance: constant(1.0)
classifier/weight: normal(0, 1.4142135623730951 / fan_out), shape=(512, 1000), axes=FanAxes(in_axis=-2, out_axis=-1, batch_axis=())
classifier/bias: constant(0.0)