100] 0 Linear-4 [-1, 50] 5,050 ReLU 100] 0 Linear-4 [-1, 10] 1,010 ReLU
, 27] 0 Conv2d-4 [-1, 256, 27, 27] 614,656 ReLU
32,768 BatchNorm2D-13 [[1, 128, 56, 56]] [1, 128, 56, 56] 512 ReLU
0 Linear-9 [[1, 784]] [1, 512] 401,920 ReLU