512] 9,408 BatchNorm2d-2 [-1, 64, 194, 512] 128 ReLU
112, 112] 9,408 BatchNorm2d-2 [-1, 64, 112, 112] 128 ReLU
16,384 BatchNorm2D-6 [[1, 64, 56, 56]] [1, 64, 56, 56] 256 ReLU
', [-1, 64, 32, 32]), ('trainable', False), ('nb_params', 128)])), ('ReLU