# Codeblock 12 Output ========================================================================================== Layer (type:depth-idx) Output Shape Param # ========================================================================================== MobileNetV2 [1, 1000] -- ├─Conv: 1-1 [1, 32, 112, 112] -- │ └─Conv2d: 2-1 [1, 32, 112, 112] 864 │ └─BatchNorm2d: 2-2 [1, 32, 112, 112] 64 │ └─ReLU6: 2-3 [1, 32, 112, 112] -- ├─InvResidualS1: 1-2 [1, 16, 112, 112] -- │ └─Conv2d: 2-4 [1, 32, 112, 112] 1,024 │ └─BatchNorm2d: 2-5 [1, 32, 112, 112] 64 │ └─ReLU6: 2-6 [1, 32, 112, 112] -- │ └─Conv2d: 2-7 [1, 32, 112, 112] 288 │ └─BatchNorm2d: 2-8 [1, 32, 112, 112] 64 │ └─ReLU6: 2-9 [1, 32, 112, 112] -- │ └─Conv2d: 2-10 [1, 16, 112, 112] 512 │ └─BatchNorm2d: 2-11 [1, 16, 112, 112] 32 ├─ModuleList: 1-3 -- -- │ └─InvResidualS2: 2-12 [1, 24, 56, 56] -- │ │ └─Conv2d: 3-1 [1, 96, 112, 112] 1,536 │ │ └─BatchNorm2d: 3-2 [1, 96, 112, 112] 192 │ │ └─ReLU6: 3-3 [1, 96, 112, 112] -- │ │ └─Conv2d: 3-4 [1, 96, 56, 56] 864 │ │ └─BatchNorm2d: 3-5 [1, 96, 56, 56] 192 │ │ └─ReLU6: 3-6 [1, 96, 56, 56] -- │ │ └─Conv2d: 3-7 [1, 24, 56, 56] 2,304 │ │ └─BatchNorm2d: 3-8 [1, 24, 56, 56] 48 │ └─InvResidualS1: 2-13 [1, 24, 56, 56] -- │ │ └─Conv2d: 3-9 [1, 144, 56, 56] 3,456 │ │ └─BatchNorm2d: 3-10 [1, 144, 56, 56] 288 │ │ └─ReLU6: 3-11 [1, 144, 56, 56] -- │ │ └─Conv2d: 3-12 [1, 144, 56, 56] 1,296 │ │ └─BatchNorm2d: 3-13 [1, 144, 56, 56] 288 │ │ └─ReLU6: 3-14 [1, 144, 56, 56] -- │ │ └─Conv2d: 3-15 [1, 24, 56, 56] 3,456 │ │ └─BatchNorm2d: 3-16 [1, 24, 56, 56] 48 ├─ModuleList: 1-4 -- -- │ └─InvResidualS2: 2-14 [1, 32, 28, 28] -- │ │ └─Conv2d: 3-17 [1, 144, 56, 56] 3,456 │ │ └─BatchNorm2d: 3-18 [1, 144, 56, 56] 288 │ │ └─ReLU6: 3-19 [1, 144, 56, 56] -- │ │ └─Conv2d: 3-20 [1, 144, 28, 28] 1,296 │ │ └─BatchNorm2d: 3-21 [1, 144, 28, 28] 288 │ │ └─ReLU6: 3-22 [1, 144, 28, 28] -- │ │ └─Conv2d: 3-23 [1, 32, 28, 28] 4,608 │ │ └─BatchNorm2d: 3-24 [1, 32, 28, 28] 64 │ └─InvResidualS1: 2-15 [1, 32, 28, 28] -- │ │ └─Conv2d: 3-25 [1, 192, 28, 28] 6,144 │ │ └─BatchNorm2d: 3-26 [1, 192, 28, 28] 384 │ │ └─ReLU6: 3-27 [1, 192, 28, 28] -- │ │ └─Conv2d: 3-28 [1, 192, 28, 28] 1,728 │ │ └─BatchNorm2d: 3-29 [1, 192, 28, 28] 384 │ │ └─ReLU6: 3-30 [1, 192, 28, 28] -- │ │ └─Conv2d: 3-31 [1, 32, 28, 28] 6,144 │ │ └─BatchNorm2d: 3-32 [1, 32, 28, 28] 64 │ └─InvResidualS1: 2-16 [1, 32, 28, 28] -- │ │ └─Conv2d: 3-33 [1, 192, 28, 28] 6,144 │ │ └─BatchNorm2d: 3-34 [1, 192, 28, 28] 384 │ │ └─ReLU6: 3-35 [1, 192, 28, 28] -- │ │ └─Conv2d: 3-36 [1, 192, 28, 28] 1,728 │ │ └─BatchNorm2d: 3-37 [1, 192, 28, 28] 384 │ │ └─ReLU6: 3-38 [1, 192, 28, 28] -- │ │ └─Conv2d: 3-39 [1, 32, 28, 28] 6,144 │ │ └─BatchNorm2d: 3-40 [1, 32, 28, 28] 64 ├─ModuleList: 1-5 -- -- │ └─InvResidualS2: 2-17 [1, 64, 14, 14] -- │ │ └─Conv2d: 3-41 [1, 192, 28, 28] 6,144 │ │ └─BatchNorm2d: 3-42 [1, 192, 28, 28] 384 │ │ └─ReLU6: 3-43 [1, 192, 28, 28] -- │ │ └─Conv2d: 3-44 [1, 192, 14, 14] 1,728 │ │ └─BatchNorm2d: 3-45 [1, 192, 14, 14] 384 │ │ └─ReLU6: 3-46 [1, 192, 14, 14] -- │ │ └─Conv2d: 3-47 [1, 64, 14, 14] 12,288 │ │ └─BatchNorm2d: 3-48 [1, 64, 14, 14] 128 │ └─InvResidualS1: 2-18 [1, 64, 14, 14] -- │ │ └─Conv2d: 3-49 [1, 384, 14, 14] 24,576 │ │ └─BatchNorm2d: 3-50 [1, 384, 14, 14] 768 │ │ └─ReLU6: 3-51 [1, 384, 14, 14] -- │ │ └─Conv2d: 3-52 [1, 384, 14, 14] 3,456 │ │ └─BatchNorm2d: 3-53 [1, 384, 14, 14] 768 │ │ └─ReLU6: 3-54 [1, 384, 14, 14] -- │ │ └─Conv2d: 3-55 [1, 64, 14, 14] 24,576 │ │ └─BatchNorm2d: 3-56 [1, 64, 14, 14] 128 │ └─InvResidualS1: 2-19 [1, 64, 14, 14] -- │ │ └─Conv2d: 3-57 [1, 384, 14, 14] 24,576 │ │ └─BatchNorm2d: 3-58 [1, 384, 14, 14] 768 │ │ └─ReLU6: 3-59 [1, 384, 14, 14] -- │ │ └─Conv2d: 3-60 [1, 384, 14, 14] 3,456 │ │ └─BatchNorm2d: 3-61 [1, 384, 14, 14] 768 │ │ └─ReLU6: 3-62 [1, 384, 14, 14] -- │ │ └─Conv2d: 3-63 [1, 64, 14, 14] 24,576 │ │ └─BatchNorm2d: 3-64 [1, 64, 14, 14] 128 │ └─InvResidualS1: 2-20 [1, 64, 14, 14] -- │ │ └─Conv2d: 3-65 [1, 384, 14, 14] 24,576 │ │ └─BatchNorm2d: 3-66 [1, 384, 14, 14] 768 │ │ └─ReLU6: 3-67 [1, 384, 14, 14] -- │ │ └─Conv2d: 3-68 [1, 384, 14, 14] 3,456 │ │ └─BatchNorm2d: 3-69 [1, 384, 14, 14] 768 │ │ └─ReLU6: 3-70 [1, 384, 14, 14] -- │ │ └─Conv2d: 3-71 [1, 64, 14, 14] 24,576 │ │ └─BatchNorm2d: 3-72 [1, 64, 14, 14] 128 ├─ModuleList: 1-6 -- -- │ └─InvResidualS1: 2-21 [1, 96, 14, 14] -- │ │ └─Conv2d: 3-73 [1, 384, 14, 14] 24,576 │ │ └─BatchNorm2d: 3-74 [1, 384, 14, 14] 768 │ │ └─ReLU6: 3-75 [1, 384, 14, 14] -- │ │ └─Conv2d: 3-76 [1, 384, 14, 14] 3,456 │ │ └─BatchNorm2d: 3-77 [1, 384, 14, 14] 768 │ │ └─ReLU6: 3-78 [1, 384, 14, 14] -- │ │ └─Conv2d: 3-79 [1, 96, 14, 14] 36,864 │ │ └─BatchNorm2d: 3-80 [1, 96, 14, 14] 192 │ └─InvResidualS1: 2-22 [1, 96, 14, 14] -- │ │ └─Conv2d: 3-81 [1, 576, 14, 14] 55,296 │ │ └─BatchNorm2d: 3-82 [1, 576, 14, 14] 1,152 │ │ └─ReLU6: 3-83 [1, 576, 14, 14] -- │ │ └─Conv2d: 3-84 [1, 576, 14, 14] 5,184 │ │ └─BatchNorm2d: 3-85 [1, 576, 14, 14] 1,152 │ │ └─ReLU6: 3-86 [1, 576, 14, 14] -- │ │ └─Conv2d: 3-87 [1, 96, 14, 14] 55,296 │ │ └─BatchNorm2d: 3-88 [1, 96, 14, 14] 192 │ └─InvResidualS1: 2-23 [1, 96, 14, 14] -- │ │ └─Conv2d: 3-89 [1, 576, 14, 14] 55,296 │ │ └─BatchNorm2d: 3-90 [1, 576, 14, 14] 1,152 │ │ └─ReLU6: 3-91 [1, 576, 14, 14] -- │ │ └─Conv2d: 3-92 [1, 576, 14, 14] 5,184 │ │ └─BatchNorm2d: 3-93 [1, 576, 14, 14] 1,152 │ │ └─ReLU6: 3-94 [1, 576, 14, 14] -- │ │ └─Conv2d: 3-95 [1, 96, 14, 14] 55,296 │ │ └─BatchNorm2d: 3-96 [1, 96, 14, 14] 192 ├─ModuleList: 1-7 -- -- │ └─InvResidualS2: 2-24 [1, 160, 7, 7] -- │ │ └─Conv2d: 3-97 [1, 576, 14, 14] 55,296 │ │ └─BatchNorm2d: 3-98 [1, 576, 14, 14] 1,152 │ │ └─ReLU6: 3-99 [1, 576, 14, 14] -- │ │ └─Conv2d: 3-100 [1, 576, 7, 7] 5,184 │ │ └─BatchNorm2d: 3-101 [1, 576, 7, 7] 1,152 │ │ └─ReLU6: 3-102 [1, 576, 7, 7] -- │ │ └─Conv2d: 3-103 [1, 160, 7, 7] 92,160 │ │ └─BatchNorm2d: 3-104 [1, 160, 7, 7] 320 │ └─InvResidualS1: 2-25 [1, 160, 7, 7] -- │ │ └─Conv2d: 3-105 [1, 960, 7, 7] 153,600 │ │ └─BatchNorm2d: 3-106 [1, 960, 7, 7] 1,920 │ │ └─ReLU6: 3-107 [1, 960, 7, 7] -- │ │ └─Conv2d: 3-108 [1, 960, 7, 7] 8,640 │ │ └─BatchNorm2d: 3-109 [1, 960, 7, 7] 1,920 │ │ └─ReLU6: 3-110 [1, 960, 7, 7] -- │ │ └─Conv2d: 3-111 [1, 160, 7, 7] 153,600 │ │ └─BatchNorm2d: 3-112 [1, 160, 7, 7] 320 │ └─InvResidualS1: 2-26 [1, 160, 7, 7] -- │ │ └─Conv2d: 3-113 [1, 960, 7, 7] 153,600 │ │ └─BatchNorm2d: 3-114 [1, 960, 7, 7] 1,920 │ │ └─ReLU6: 3-115 [1, 960, 7, 7] -- │ │ └─Conv2d: 3-116 [1, 960, 7, 7] 8,640 │ │ └─BatchNorm2d: 3-117 [1, 960, 7, 7] 1,920 │ │ └─ReLU6: 3-118 [1, 960, 7, 7] -- │ │ └─Conv2d: 3-119 [1, 160, 7, 7] 153,600 │ │ └─BatchNorm2d: 3-120 [1, 160, 7, 7] 320 ├─InvResidualS1: 1-8 [1, 320, 7, 7] -- │ └─Conv2d: 2-27 [1, 960, 7, 7] 153,600 │ └─BatchNorm2d: 2-28 [1, 960, 7, 7] 1,920 │ └─ReLU6: 2-29 [1, 960, 7, 7] -- │ └─Conv2d: 2-30 [1, 960, 7, 7] 8,640 │ └─BatchNorm2d: 2-31 [1, 960, 7, 7] 1,920 │ └─ReLU6: 2-32 [1, 960, 7, 7] -- │ └─Conv2d: 2-33 [1, 320, 7, 7] 307,200 │ └─BatchNorm2d: 2-34 [1, 320, 7, 7] 640 ├─Conv: 1-9 [1, 1280, 7, 7] -- │ └─Conv2d: 2-35 [1, 1280, 7, 7] 409,600 │ └─BatchNorm2d: 2-36 [1, 1280, 7, 7] 2,560 │ └─ReLU6: 2-37 [1, 1280, 7, 7] -- ├─AdaptiveAvgPool2d: 1-10 [1, 1280, 1, 1] -- ├─Dropout: 1-11 [1, 1280] -- ├─Linear: 1-12 [1, 1000] 1,281,000 ========================================================================================== Total params: 3,505,960 Trainable params: 3,505,960 Non-trainable params: 0 Total mult-adds (Units.MEGABYTES): 313.65 ========================================================================================== Input size (MB): 0.60 Forward/backward pass size (MB): 113.28 Params size (MB): 14.02 Estimated Total Size (MB): 127.91 ==========================================================================================