224] 0 Conv2d-3 [-1, 64, 224, 224] 36,928 ReLU
196] 256 MaxPool1d-3 [-1, 16, 98] 0 ReLU
196] 256 MaxPool1d-3 [-1, 16, 98] 0 ReLU
, 20] 0 Linear-3 [-1, 15] 315 ReLU
224] 0 Conv2d-3 [-1, 64, 224, 224] 36,928 ReLU
196] 256 MaxPool1d-3 [-1, 16, 98] 0 ReLU
16,384 BatchNorm2D-9 [[1, 64, 56, 56]] [1, 64, 56, 56] 256 ReLU
, 16] 0 Conv2d-3 [-1, 1, 28, 28] 10 ReLU