, 256] 36,864 BatchNorm2d-9 [-1, 64, 97, 256] 128 ReLU
13, 13] 0 Conv2d-9 [-1, 384, 13, 13] 1,327,488 ReLU
128] 40,064 BatchNorm1d-9 [-1, 128] 256 ReLU
262,144 BatchNorm2D-29 [[1, 256, 14, 14]] [1, 256, 14, 14] 1,024 ReLU