# Codeblock 9 Output ========================================================================================== Layer (type:depth-idx) Output Shape Param # ========================================================================================== ResNeXt [1000] -- ├─Conv2d: 1-1 [1, 64, 112, 112] 9,408 ├─BatchNorm2d: 1-2 [1, 64, 112, 112] 128 ├─ReLU: 1-3 [1, 64, 112, 112] -- ├─MaxPool2d: 1-4 [1, 64, 56, 56] -- ├─ModuleList: 1-5 -- -- │ └─Block: 2-1 [1, 256, 56, 56] -- │ │ └─Conv2d: 3-1 [1, 256, 56, 56] 16,384 │ │ └─BatchNorm2d: 3-2 [1, 256, 56, 56] 512 │ │ └─Conv2d: 3-3 [1, 128, 56, 56] 8,192 │ │ └─BatchNorm2d: 3-4 [1, 128, 56, 56] 256 │ │ └─ReLU: 3-5 [1, 128, 56, 56] -- │ │ └─Conv2d: 3-6 [1, 128, 56, 56] 4,608 │ │ └─BatchNorm2d: 3-7 [1, 128, 56, 56] 256 │ │ └─ReLU: 3-8 [1, 128, 56, 56] -- │ │ └─Conv2d: 3-9 [1, 256, 56, 56] 32,768 │ │ └─BatchNorm2d: 3-10 [1, 256, 56, 56] 512 │ │ └─ReLU: 3-11 [1, 256, 56, 56] -- │ └─Block: 2-2 [1, 256, 56, 56] -- │ │ └─Conv2d: 3-12 [1, 128, 56, 56] 32,768 │ │ └─BatchNorm2d: 3-13 [1, 128, 56, 56] 256 │ │ └─ReLU: 3-14 [1, 128, 56, 56] -- │ │ └─Conv2d: 3-15 [1, 128, 56, 56] 4,608 │ │ └─BatchNorm2d: 3-16 [1, 128, 56, 56] 256 │ │ └─ReLU: 3-17 [1, 128, 56, 56] -- │ │ └─Conv2d: 3-18 [1, 256, 56, 56] 32,768 │ │ └─BatchNorm2d: 3-19 [1, 256, 56, 56] 512 │ │ └─ReLU: 3-20 [1, 256, 56, 56] -- │ └─Block: 2-3 [1, 256, 56, 56] -- │ │ └─Conv2d: 3-21 [1, 128, 56, 56] 32,768 │ │ └─BatchNorm2d: 3-22 [1, 128, 56, 56] 256 │ │ └─ReLU: 3-23 [1, 128, 56, 56] -- │ │ └─Conv2d: 3-24 [1, 128, 56, 56] 4,608 │ │ └─BatchNorm2d: 3-25 [1, 128, 56, 56] 256 │ │ └─ReLU: 3-26 [1, 128, 56, 56] -- │ │ └─Conv2d: 3-27 [1, 256, 56, 56] 32,768 │ │ └─BatchNorm2d: 3-28 [1, 256, 56, 56] 512 │ │ └─ReLU: 3-29 [1, 256, 56, 56] -- ├─ModuleList: 1-6 -- -- │ └─Block: 2-4 [1, 512, 28, 28] -- │ │ └─Conv2d: 3-30 [1, 512, 28, 28] 131,072 │ │ └─BatchNorm2d: 3-31 [1, 512, 28, 28] 1,024 │ │ └─Conv2d: 3-32 [1, 256, 56, 56] 65,536 │ │ └─BatchNorm2d: 3-33 [1, 256, 56, 56] 512 │ │ └─ReLU: 3-34 [1, 256, 56, 56] -- │ │ └─Conv2d: 3-35 [1, 256, 28, 28] 18,432 │ │ └─BatchNorm2d: 3-36 [1, 256, 28, 28] 512 │ │ └─ReLU: 3-37 [1, 256, 28, 28] -- │ │ └─Conv2d: 3-38 [1, 512, 28, 28] 131,072 │ │ └─BatchNorm2d: 3-39 [1, 512, 28, 28] 1,024 │ │ └─ReLU: 3-40 [1, 512, 28, 28] -- │ └─Block: 2-5 [1, 512, 28, 28] -- │ │ └─Conv2d: 3-41 [1, 256, 28, 28] 131,072 │ │ └─BatchNorm2d: 3-42 [1, 256, 28, 28] 512 │ │ └─ReLU: 3-43 [1, 256, 28, 28] -- │ │ └─Conv2d: 3-44 [1, 256, 28, 28] 18,432 │ │ └─BatchNorm2d: 3-45 [1, 256, 28, 28] 512 │ │ └─ReLU: 3-46 [1, 256, 28, 28] -- │ │ └─Conv2d: 3-47 [1, 512, 28, 28] 131,072 │ │ └─BatchNorm2d: 3-48 [1, 512, 28, 28] 1,024 │ │ └─ReLU: 3-49 [1, 512, 28, 28] -- │ └─Block: 2-6 [1, 512, 28, 28] -- │ │ └─Conv2d: 3-50 [1, 256, 28, 28] 131,072 │ │ └─BatchNorm2d: 3-51 [1, 256, 28, 28] 512 │ │ └─ReLU: 3-52 [1, 256, 28, 28] -- │ │ └─Conv2d: 3-53 [1, 256, 28, 28] 18,432 │ │ └─BatchNorm2d: 3-54 [1, 256, 28, 28] 512 │ │ └─ReLU: 3-55 [1, 256, 28, 28] -- │ │ └─Conv2d: 3-56 [1, 512, 28, 28] 131,072 │ │ └─BatchNorm2d: 3-57 [1, 512, 28, 28] 1,024 │ │ └─ReLU: 3-58 [1, 512, 28, 28] -- │ └─Block: 2-7 [1, 512, 28, 28] -- │ │ └─Conv2d: 3-59 [1, 256, 28, 28] 131,072 │ │ └─BatchNorm2d: 3-60 [1, 256, 28, 28] 512 │ │ └─ReLU: 3-61 [1, 256, 28, 28] -- │ │ └─Conv2d: 3-62 [1, 256, 28, 28] 18,432 │ │ └─BatchNorm2d: 3-63 [1, 256, 28, 28] 512 │ │ └─ReLU: 3-64 [1, 256, 28, 28] -- │ │ └─Conv2d: 3-65 [1, 512, 28, 28] 131,072 │ │ └─BatchNorm2d: 3-66 [1, 512, 28, 28] 1,024 │ │ └─ReLU: 3-67 [1, 512, 28, 28] -- ├─ModuleList: 1-7 -- -- │ └─Block: 2-8 [1, 1024, 14, 14] -- │ │ └─Conv2d: 3-68 [1, 1024, 14, 14] 524,288 │ │ └─BatchNorm2d: 3-69 [1, 1024, 14, 14] 2,048 │ │ └─Conv2d: 3-70 [1, 512, 28, 28] 262,144 │ │ └─BatchNorm2d: 3-71 [1, 512, 28, 28] 1,024 │ │ └─ReLU: 3-72 [1, 512, 28, 28] -- │ │ └─Conv2d: 3-73 [1, 512, 14, 14] 73,728 │ │ └─BatchNorm2d: 3-74 [1, 512, 14, 14] 1,024 │ │ └─ReLU: 3-75 [1, 512, 14, 14] -- │ │ └─Conv2d: 3-76 [1, 1024, 14, 14] 524,288 │ │ └─BatchNorm2d: 3-77 [1, 1024, 14, 14] 2,048 │ │ └─ReLU: 3-78 [1, 1024, 14, 14] -- │ └─Block: 2-9 [1, 1024, 14, 14] -- │ │ └─Conv2d: 3-79 [1, 512, 14, 14] 524,288 │ │ └─BatchNorm2d: 3-80 [1, 512, 14, 14] 1,024 │ │ └─ReLU: 3-81 [1, 512, 14, 14] -- │ │ └─Conv2d: 3-82 [1, 512, 14, 14] 73,728 │ │ └─BatchNorm2d: 3-83 [1, 512, 14, 14] 1,024 │ │ └─ReLU: 3-84 [1, 512, 14, 14] -- │ │ └─Conv2d: 3-85 [1, 1024, 14, 14] 524,288 │ │ └─BatchNorm2d: 3-86 [1, 1024, 14, 14] 2,048 │ │ └─ReLU: 3-87 [1, 1024, 14, 14] -- │ └─Block: 2-10 [1, 1024, 14, 14] -- │ │ └─Conv2d: 3-88 [1, 512, 14, 14] 524,288 │ │ └─BatchNorm2d: 3-89 [1, 512, 14, 14] 1,024 │ │ └─ReLU: 3-90 [1, 512, 14, 14] -- │ │ └─Conv2d: 3-91 [1, 512, 14, 14] 73,728 │ │ └─BatchNorm2d: 3-92 [1, 512, 14, 14] 1,024 │ │ └─ReLU: 3-93 [1, 512, 14, 14] -- │ │ └─Conv2d: 3-94 [1, 1024, 14, 14] 524,288 │ │ └─BatchNorm2d: 3-95 [1, 1024, 14, 14] 2,048 │ │ └─ReLU: 3-96 [1, 1024, 14, 14] -- │ └─Block: 2-11 [1, 1024, 14, 14] -- │ │ └─Conv2d: 3-97 [1, 512, 14, 14] 524,288 │ │ └─BatchNorm2d: 3-98 [1, 512, 14, 14] 1,024 │ │ └─ReLU: 3-99 [1, 512, 14, 14] -- │ │ └─Conv2d: 3-100 [1, 512, 14, 14] 73,728 │ │ └─BatchNorm2d: 3-101 [1, 512, 14, 14] 1,024 │ │ └─ReLU: 3-102 [1, 512, 14, 14] -- │ │ └─Conv2d: 3-103 [1, 1024, 14, 14] 524,288 │ │ └─BatchNorm2d: 3-104 [1, 1024, 14, 14] 2,048 │ │ └─ReLU: 3-105 [1, 1024, 14, 14] -- │ └─Block: 2-12 [1, 1024, 14, 14] -- │ │ └─Conv2d: 3-106 [1, 512, 14, 14] 524,288 │ │ └─BatchNorm2d: 3-107 [1, 512, 14, 14] 1,024 │ │ └─ReLU: 3-108 [1, 512, 14, 14] -- │ │ └─Conv2d: 3-109 [1, 512, 14, 14] 73,728 │ │ └─BatchNorm2d: 3-110 [1, 512, 14, 14] 1,024 │ │ └─ReLU: 3-111 [1, 512, 14, 14] -- │ │ └─Conv2d: 3-112 [1, 1024, 14, 14] 524,288 │ │ └─BatchNorm2d: 3-113 [1, 1024, 14, 14] 2,048 │ │ └─ReLU: 3-114 [1, 1024, 14, 14] -- │ └─Block: 2-13 [1, 1024, 14, 14] -- │ │ └─Conv2d: 3-115 [1, 512, 14, 14] 524,288 │ │ └─BatchNorm2d: 3-116 [1, 512, 14, 14] 1,024 │ │ └─ReLU: 3-117 [1, 512, 14, 14] -- │ │ └─Conv2d: 3-118 [1, 512, 14, 14] 73,728 │ │ └─BatchNorm2d: 3-119 [1, 512, 14, 14] 1,024 │ │ └─ReLU: 3-120 [1, 512, 14, 14] -- │ │ └─Conv2d: 3-121 [1, 1024, 14, 14] 524,288 │ │ └─BatchNorm2d: 3-122 [1, 1024, 14, 14] 2,048 │ │ └─ReLU: 3-123 [1, 1024, 14, 14] -- ├─ModuleList: 1-8 -- -- │ └─Block: 2-14 [1, 2048, 7, 7] -- │ │ └─Conv2d: 3-124 [1, 2048, 7, 7] 2,097,152 │ │ └─BatchNorm2d: 3-125 [1, 2048, 7, 7] 4,096 │ │ └─Conv2d: 3-126 [1, 1024, 14, 14] 1,048,576 │ │ └─BatchNorm2d: 3-127 [1, 1024, 14, 14] 2,048 │ │ └─ReLU: 3-128 [1, 1024, 14, 14] -- │ │ └─Conv2d: 3-129 [1, 1024, 7, 7] 294,912 │ │ └─BatchNorm2d: 3-130 [1, 1024, 7, 7] 2,048 │ │ └─ReLU: 3-131 [1, 1024, 7, 7] -- │ │ └─Conv2d: 3-132 [1, 2048, 7, 7] 2,097,152 │ │ └─BatchNorm2d: 3-133 [1, 2048, 7, 7] 4,096 │ │ └─ReLU: 3-134 [1, 2048, 7, 7] -- │ └─Block: 2-15 [1, 2048, 7, 7] -- │ │ └─Conv2d: 3-135 [1, 1024, 7, 7] 2,097,152 │ │ └─BatchNorm2d: 3-136 [1, 1024, 7, 7] 2,048 │ │ └─ReLU: 3-137 [1, 1024, 7, 7] -- │ │ └─Conv2d: 3-138 [1, 1024, 7, 7] 294,912 │ │ └─BatchNorm2d: 3-139 [1, 1024, 7, 7] 2,048 │ │ └─ReLU: 3-140 [1, 1024, 7, 7] -- │ │ └─Conv2d: 3-141 [1, 2048, 7, 7] 2,097,152 │ │ └─BatchNorm2d: 3-142 [1, 2048, 7, 7] 4,096 │ │ └─ReLU: 3-143 [1, 2048, 7, 7] -- │ └─Block: 2-16 [1, 2048, 7, 7] -- │ │ └─Conv2d: 3-144 [1, 1024, 7, 7] 2,097,152 │ │ └─BatchNorm2d: 3-145 [1, 1024, 7, 7] 2,048 │ │ └─ReLU: 3-146 [1, 1024, 7, 7] -- │ │ └─Conv2d: 3-147 [1, 1024, 7, 7] 294,912 │ │ └─BatchNorm2d: 3-148 [1, 1024, 7, 7] 2,048 │ │ └─ReLU: 3-149 [1, 1024, 7, 7] -- │ │ └─Conv2d: 3-150 [1, 2048, 7, 7] 2,097,152 │ │ └─BatchNorm2d: 3-151 [1, 2048, 7, 7] 4,096 │ │ └─ReLU: 3-152 [1, 2048, 7, 7] -- ├─AdaptiveAvgPool2d: 1-9 [1, 2048, 1, 1] -- ├─Linear: 1-10 [1, 1000] 2,049,000 ========================================================================================== Total params: 25,028,904 Trainable params: 25,028,904 Non-trainable params: 0 Total mult-adds (Units.GIGABYTES): 6.28 ========================================================================================== Input size (MB): 0.60 Forward/backward pass size (MB): 230.42 Params size (MB): 100.12 Estimated Total Size (MB): 331.13 ==========================================================================================