===================== Linear-1 [-1, 100] 100,100 ReLU ===================== Linear-1 [-1, 100] 19,660,900 ReLU
===================== Conv2d-1 [-1, 64, 224, 224] 1,792 ReLU
===================== Linear-1 [-1, 20] 320 ReLU
===================== Conv2d-1 [-1, 64, 224, 224] 1,792 ReLU
===================== Conv2d-1 [-1, 96, 55, 55] 34,944 ReLU
Conv1d-3 [[1, 512, 98]] [1, 512, 98] 0 ReLU
4,096 BatchNorm2D-3 [[1, 64, 56, 56]] [1, 64, 56, 56] 256 ReLU
===================== Conv2d-1 [-1, 1, 16, 16] 10 ReLU