112] 0 Conv2d-6 [-1, 128, 112, 112] 73,856 ReLU
256] 36,864 BatchNorm2d-6 [-1, 64, 97, 256] 128 ReLU
, 97] 4,224 MaxPool1d-6 [-1, 128, 48] 0 ReLU
, 97] 4,224 MaxPool1d-6 [-1, 128, 48] 0 ReLU
112] 0 Conv2d-6 [-1, 128, 112, 112] 73,856 ReLU
, 97] 4,224 MaxPool1d-6 [-1, 128, 48] 0 ReLU
65,536 BatchNorm2D-19 [[1, 128, 28, 28]] [1, 128, 28, 28] 512 ReLU