backbone/stem/conv1/weight: normal(0, 1.4142135623730951 / fan_out), shape=[7, 7, 3, 4], 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)
classifier/weight: normal(0, 1.4142135623730951 / fan_out), shape=(4, 1000), axes=FanAxes(in_axis=-2, out_axis=-1, batch_axis=())
classifier/bias: constant(0.0)