------------------------------------------------------------------------------------------------------- Layer (type) Output Shape Param # Tr. Param # ======================================================================================================= Conv3d-1 [1, 7, 224, 304, 304] 1,757 1,757 BatchNorm3d-2 [1, 7, 224, 304, 304] 14 14 ReLU-3 [1, 7, 224, 304, 304] 0 0 Conv3d-4 [1, 7, 224, 304, 304] 6,132 6,132 BatchNorm3d-5 [1, 7, 224, 304, 304] 14 14 ReLU-6 [1, 7, 224, 304, 304] 0 0 Dropout3d-7 [1, 7, 224, 304, 304] 0 0 MaxPool3d-8 [1, 7, 112, 152, 152], [1, 7, 112, 152, 152] 0 0 Conv3d-9 [1, 14, 112, 152, 152] 12,264 12,264 BatchNorm3d-10 [1, 14, 112, 152, 152] 28 28 ReLU-11 [1, 14, 112, 152, 152] 0 0 Conv3d-12 [1, 14, 112, 152, 152] 24,514 24,514 BatchNorm3d-13 [1, 14, 112, 152, 152] 28 28 ReLU-14 [1, 14, 112, 152, 152] 0 0 Dropout3d-15 [1, 14, 112, 152, 152] 0 0 Conv3d-16 [1, 28, 56, 76, 76] 49,028 49,028 BatchNorm3d-17 [1, 28, 56, 76, 76] 56 56 ReLU-18 [1, 28, 56, 76, 76] 0 0 Conv3d-19 [1, 28, 56, 76, 76] 98,028 98,028 BatchNorm3d-20 [1, 28, 56, 76, 76] 56 56 ReLU-21 [1, 28, 56, 76, 76] 0 0 Dropout3d-22 [1, 28, 56, 76, 76] 0 0 Conv3d-23 [1, 56, 28, 38, 38] 196,056 196,056 BatchNorm3d-24 [1, 56, 28, 38, 38] 112 112 ReLU-25 [1, 56, 28, 38, 38] 0 0 Conv3d-26 [1, 56, 28, 38, 38] 392,056 392,056 BatchNorm3d-27 [1, 56, 28, 38, 38] 112 112 ReLU-28 [1, 56, 28, 38, 38] 0 0 Dropout3d-29 [1, 56, 28, 38, 38] 0 0 Conv3d-30 [1, 112, 14, 19, 19] 784,112 784,112 BatchNorm3d-31 [1, 112, 14, 19, 19] 224 224 ReLU-32 [1, 112, 14, 19, 19] 0 0 Conv3d-33 [1, 112, 14, 19, 19] 1,568,112 1,568,112 BatchNorm3d-34 [1, 112, 14, 19, 19] 224 224 ReLU-35 [1, 112, 14, 19, 19] 0 0 Dropout3d-36 [1, 112, 14, 19, 19] 0 0 ConvTranspose3d-37 [1, 112, 28, 38, 38] 100,464 100,464 Conv3d-38 [1, 56, 28, 38, 38] 784,056 784,056 BatchNorm3d-39 [1, 56, 28, 38, 38] 112 112 ReLU-40 [1, 56, 28, 38, 38] 0 0 Conv3d-41 [1, 56, 28, 38, 38] 392,056 392,056 BatchNorm3d-42 [1, 56, 28, 38, 38] 112 112 ReLU-43 [1, 56, 28, 38, 38] 0 0 Dropout3d-44 [1, 56, 28, 38, 38] 0 0 ConvTranspose3d-45 [1, 112, 56, 76, 76] 100,464 100,464 Conv3d-46 [1, 28, 56, 76, 76] 392,028 392,028 BatchNorm3d-47 [1, 28, 56, 76, 76] 56 56 ReLU-48 [1, 28, 56, 76, 76] 0 0 Conv3d-49 [1, 28, 56, 76, 76] 98,028 98,028 BatchNorm3d-50 [1, 28, 56, 76, 76] 56 56 ReLU-51 [1, 28, 56, 76, 76] 0 0 Dropout3d-52 [1, 28, 56, 76, 76] 0 0 ConvTranspose3d-53 [1, 56, 112, 152, 152] 25,144 25,144 Conv3d-54 [1, 14, 112, 152, 152] 98,014 98,014 BatchNorm3d-55 [1, 14, 112, 152, 152] 28 28 ReLU-56 [1, 14, 112, 152, 152] 0 0 Conv3d-57 [1, 14, 112, 152, 152] 24,514 24,514 BatchNorm3d-58 [1, 14, 112, 152, 152] 28 28 ReLU-59 [1, 14, 112, 152, 152] 0 0 Dropout3d-60 [1, 14, 112, 152, 152] 0 0 ConvTranspose3d-61 [1, 28, 224, 304, 304] 6,300 6,300 Conv3d-62 [1, 7, 224, 304, 304] 24,507 24,507 BatchNorm3d-63 [1, 7, 224, 304, 304] 14 14 ReLU-64 [1, 7, 224, 304, 304] 0 0 Conv3d-65 [1, 7, 224, 304, 304] 6,132 6,132 BatchNorm3d-66 [1, 7, 224, 304, 304] 14 14 ReLU-67 [1, 7, 224, 304, 304] 0 0 Dropout3d-68 [1, 7, 224, 304, 304] 0 0 Conv3d-69 [1, 2, 224, 304, 304] 30 30 ======================================================================================================= Total params: 5,185,084 Trainable params: 5,185,084 Non-trainable params: 0 ------------------------------------------------------------------------------------------------------- Process finished with exit code 0