------------------------------------------------------------------------------------------------------- Layer (type) Output Shape Param # Tr. Param # ======================================================================================================= Conv3d-1 [1, 8, 224, 304, 304] 1,008 1,008 BatchNorm3d-2 [1, 8, 224, 304, 304] 16 16 ReLU-3 [1, 8, 224, 304, 304] 0 0 Conv3d-4 [1, 8, 224, 304, 304] 8,008 8,008 BatchNorm3d-5 [1, 8, 224, 304, 304] 16 16 ReLU-6 [1, 8, 224, 304, 304] 0 0 Dropout3d-7 [1, 8, 224, 304, 304] 0 0 MaxPool3d-8 [1, 8, 112, 152, 152], [1, 8, 112, 152, 152] 0 0 Conv3d-9 [1, 16, 112, 152, 152] 16,016 16,016 BatchNorm3d-10 [1, 16, 112, 152, 152] 32 32 ReLU-11 [1, 16, 112, 152, 152] 0 0 Conv3d-12 [1, 16, 112, 152, 152] 32,016 32,016 BatchNorm3d-13 [1, 16, 112, 152, 152] 32 32 ReLU-14 [1, 16, 112, 152, 152] 0 0 Dropout3d-15 [1, 16, 112, 152, 152] 0 0 Conv3d-16 [1, 32, 56, 76, 76] 64,032 64,032 BatchNorm3d-17 [1, 32, 56, 76, 76] 64 64 ReLU-18 [1, 32, 56, 76, 76] 0 0 Conv3d-19 [1, 32, 56, 76, 76] 128,032 128,032 BatchNorm3d-20 [1, 32, 56, 76, 76] 64 64 ReLU-21 [1, 32, 56, 76, 76] 0 0 Dropout3d-22 [1, 32, 56, 76, 76] 0 0 Conv3d-23 [1, 64, 28, 38, 38] 256,064 256,064 BatchNorm3d-24 [1, 64, 28, 38, 38] 128 128 ReLU-25 [1, 64, 28, 38, 38] 0 0 Conv3d-26 [1, 64, 28, 38, 38] 512,064 512,064 BatchNorm3d-27 [1, 64, 28, 38, 38] 128 128 ReLU-28 [1, 64, 28, 38, 38] 0 0 Dropout3d-29 [1, 64, 28, 38, 38] 0 0 Conv3d-30 [1, 128, 14, 19, 19] 1,024,128 1,024,128 BatchNorm3d-31 [1, 128, 14, 19, 19] 256 256 ReLU-32 [1, 128, 14, 19, 19] 0 0 Conv3d-33 [1, 128, 14, 19, 19] 2,048,128 2,048,128 BatchNorm3d-34 [1, 128, 14, 19, 19] 256 256 ReLU-35 [1, 128, 14, 19, 19] 0 0 Dropout3d-36 [1, 128, 14, 19, 19] 0 0 ConvTranspose3d-37 [1, 128, 28, 38, 38] 131,200 131,200 Conv3d-38 [1, 64, 28, 38, 38] 1,024,064 1,024,064 BatchNorm3d-39 [1, 64, 28, 38, 38] 128 128 ReLU-40 [1, 64, 28, 38, 38] 0 0 Conv3d-41 [1, 64, 28, 38, 38] 512,064 512,064 BatchNorm3d-42 [1, 64, 28, 38, 38] 128 128 ReLU-43 [1, 64, 28, 38, 38] 0 0 Dropout3d-44 [1, 64, 28, 38, 38] 0 0 ConvTranspose3d-45 [1, 128, 56, 76, 76] 131,200 131,200 Conv3d-46 [1, 32, 56, 76, 76] 512,032 512,032 BatchNorm3d-47 [1, 32, 56, 76, 76] 64 64 ReLU-48 [1, 32, 56, 76, 76] 0 0 Conv3d-49 [1, 32, 56, 76, 76] 128,032 128,032 BatchNorm3d-50 [1, 32, 56, 76, 76] 64 64 ReLU-51 [1, 32, 56, 76, 76] 0 0 Dropout3d-52 [1, 32, 56, 76, 76] 0 0 ConvTranspose3d-53 [1, 64, 112, 152, 152] 32,832 32,832 Conv3d-54 [1, 16, 112, 152, 152] 128,016 128,016 BatchNorm3d-55 [1, 16, 112, 152, 152] 32 32 ReLU-56 [1, 16, 112, 152, 152] 0 0 Conv3d-57 [1, 16, 112, 152, 152] 32,016 32,016 BatchNorm3d-58 [1, 16, 112, 152, 152] 32 32 ReLU-59 [1, 16, 112, 152, 152] 0 0 Dropout3d-60 [1, 16, 112, 152, 152] 0 0 ConvTranspose3d-61 [1, 32, 224, 304, 304] 8,224 8,224 Conv3d-62 [1, 8, 224, 304, 304] 32,008 32,008 BatchNorm3d-63 [1, 8, 224, 304, 304] 16 16 ReLU-64 [1, 8, 224, 304, 304] 0 0 Conv3d-65 [1, 8, 224, 304, 304] 8,008 8,008 BatchNorm3d-66 [1, 8, 224, 304, 304] 16 16 ReLU-67 [1, 8, 224, 304, 304] 0 0 Dropout3d-68 [1, 8, 224, 304, 304] 0 0 Conv3d-69 [1, 2, 224, 304, 304] 34 34 ======================================================================================================= Total params: 6,770,698 Trainable params: 6,770,698 Non-trainable params: 0 ------------------------------------------------------------------------------------------------------- Process finished with exit code 0