, 64] 0 Linear-8 [-1, 32] 2,080 ReLU , 64] 0 Linear-8 [-1, 32] 2,080 ReLU , 64] 0 Linear-8 [-1, 32] 2,080 ReLU , 64] 0 Linear-8 [-1, 32] 2,080 ReLU
, 64] 0 Linear-8 [-1, 32] 2,080 ReLU , 64] 0 Linear-8 [-1, 32] 2,080 ReLU , 64] 0 Linear-8 [-1, 32] 2,080 ReLU
112] 0 Conv2d-8 [-1, 128, 112, 112] 147,584 ReLU
, 64] 0 Linear-8 [-1, 32] 2,080 ReLU
, 64] 0 Linear-8 [-1, 32] 2,080 ReLU
112] 0 Conv2d-8 [-1, 128, 112, 112] 147,584 ReLU
, 64] 0 Linear-8 [-1, 32] 2,080 ReLU
131,072 BatchNorm2D-26 [[1, 256, 28, 28]] [1, 256, 28, 28] 1,024 ReLU