====================================================================== Layer (type:depth-idx) Param # ====================================================================== smp_UNetPlusPlus -- ├─ResNetEncoder: 1-1 -- │ └─Conv2d: 2-1 3,136 │ └─BatchNorm2d: 2-2 128 │ └─ReLU: 2-3 -- │ └─MaxPool2d: 2-4 -- │ └─Sequential: 2-5 -- │ │ └─BasicBlock: 3-1 -- │ │ │ └─Conv2d: 4-1 36,864 │ │ │ └─BatchNorm2d: 4-2 128 │ │ │ └─ReLU: 4-3 -- │ │ │ └─Conv2d: 4-4 36,864 │ │ │ └─BatchNorm2d: 4-5 128 │ │ └─BasicBlock: 3-2 -- │ │ │ └─Conv2d: 4-6 36,864 │ │ │ └─BatchNorm2d: 4-7 128 │ │ │ └─ReLU: 4-8 -- │ │ │ └─Conv2d: 4-9 36,864 │ │ │ └─BatchNorm2d: 4-10 128 │ └─Sequential: 2-6 -- │ │ └─BasicBlock: 3-3 -- │ │ │ └─Conv2d: 4-11 73,728 │ │ │ └─BatchNorm2d: 4-12 256 │ │ │ └─ReLU: 4-13 -- │ │ │ └─Conv2d: 4-14 147,456 │ │ │ └─BatchNorm2d: 4-15 256 │ │ │ └─Sequential: 4-16 -- │ │ │ │ └─Conv2d: 5-1 8,192 │ │ │ │ └─BatchNorm2d: 5-2 256 │ │ └─BasicBlock: 3-4 -- │ │ │ └─Conv2d: 4-17 147,456 │ │ │ └─BatchNorm2d: 4-18 256 │ │ │ └─ReLU: 4-19 -- │ │ │ └─Conv2d: 4-20 147,456 │ │ │ └─BatchNorm2d: 4-21 256 │ └─Sequential: 2-7 -- │ │ └─BasicBlock: 3-5 -- │ │ │ └─Conv2d: 4-22 294,912 │ │ │ └─BatchNorm2d: 4-23 512 │ │ │ └─ReLU: 4-24 -- │ │ │ └─Conv2d: 4-25 589,824 │ │ │ └─BatchNorm2d: 4-26 512 │ │ │ └─Sequential: 4-27 -- │ │ │ │ └─Conv2d: 5-3 32,768 │ │ │ │ └─BatchNorm2d: 5-4 512 │ │ └─BasicBlock: 3-6 -- │ │ │ └─Conv2d: 4-28 589,824 │ │ │ └─BatchNorm2d: 4-29 512 │ │ │ └─ReLU: 4-30 -- │ │ │ └─Conv2d: 4-31 589,824 │ │ │ └─BatchNorm2d: 4-32 512 │ └─Sequential: 2-8 -- │ │ └─BasicBlock: 3-7 -- │ │ │ └─Conv2d: 4-33 1,179,648 │ │ │ └─BatchNorm2d: 4-34 1,024 │ │ │ └─ReLU: 4-35 -- │ │ │ └─Conv2d: 4-36 2,359,296 │ │ │ └─BatchNorm2d: 4-37 1,024 │ │ │ └─Sequential: 4-38 -- │ │ │ │ └─Conv2d: 5-5 131,072 │ │ │ │ └─BatchNorm2d: 5-6 1,024 │ │ └─BasicBlock: 3-8 -- │ │ │ └─Conv2d: 4-39 2,359,296 │ │ │ └─BatchNorm2d: 4-40 1,024 │ │ │ └─ReLU: 4-41 -- │ │ │ └─Conv2d: 4-42 2,359,296 │ │ │ └─BatchNorm2d: 4-43 1,024 ├─UnetPlusPlusDecoder: 1-2 -- │ └─Identity: 2-9 -- │ └─ModuleDict: 2-10 -- │ │ └─DecoderBlock: 3-9 -- │ │ │ └─Conv2dReLU: 4-44 -- │ │ │ │ └─Conv2d: 5-7 1,769,472 │ │ │ │ └─BatchNorm2d: 5-8 512 │ │ │ │ └─ReLU: 5-9 -- │ │ │ └─Attention: 4-45 -- │ │ │ │ └─Identity: 5-10 -- │ │ │ └─Conv2dReLU: 4-46 -- │ │ │ │ └─Conv2d: 5-11 589,824 │ │ │ │ └─BatchNorm2d: 5-12 512 │ │ │ │ └─ReLU: 5-13 -- │ │ │ └─Attention: 4-47 -- │ │ │ │ └─Identity: 5-14 -- │ │ └─DecoderBlock: 3-10 -- │ │ │ └─Conv2dReLU: 4-48 -- │ │ │ │ └─Conv2d: 5-15 589,824 │ │ │ │ └─BatchNorm2d: 5-16 256 │ │ │ │ └─ReLU: 5-17 -- │ │ │ └─Attention: 4-49 -- │ │ │ │ └─Identity: 5-18 -- │ │ │ └─Conv2dReLU: 4-50 -- │ │ │ │ └─Conv2d: 5-19 147,456 │ │ │ │ └─BatchNorm2d: 5-20 256 │ │ │ │ └─ReLU: 5-21 -- │ │ │ └─Attention: 4-51 -- │ │ │ │ └─Identity: 5-22 -- │ │ └─DecoderBlock: 3-11 -- │ │ │ └─Conv2dReLU: 4-52 -- │ │ │ │ └─Conv2d: 5-23 442,368 │ │ │ │ └─BatchNorm2d: 5-24 256 │ │ │ │ └─ReLU: 5-25 -- │ │ │ └─Attention: 4-53 -- │ │ │ │ └─Identity: 5-26 -- │ │ │ └─Conv2dReLU: 4-54 -- │ │ │ │ └─Conv2d: 5-27 147,456 │ │ │ │ └─BatchNorm2d: 5-28 256 │ │ │ │ └─ReLU: 5-29 -- │ │ │ └─Attention: 4-55 -- │ │ │ │ └─Identity: 5-30 -- │ │ └─DecoderBlock: 3-12 -- │ │ │ └─Conv2dReLU: 4-56 -- │ │ │ │ └─Conv2d: 5-31 184,320 │ │ │ │ └─BatchNorm2d: 5-32 128 │ │ │ │ └─ReLU: 5-33 -- │ │ │ └─Attention: 4-57 -- │ │ │ │ └─Identity: 5-34 -- │ │ │ └─Conv2dReLU: 4-58 -- │ │ │ │ └─Conv2d: 5-35 36,864 │ │ │ │ └─BatchNorm2d: 5-36 128 │ │ │ │ └─ReLU: 5-37 -- │ │ │ └─Attention: 4-59 -- │ │ │ │ └─Identity: 5-38 -- │ │ └─DecoderBlock: 3-13 -- │ │ │ └─Conv2dReLU: 4-60 -- │ │ │ │ └─Conv2d: 5-39 147,456 │ │ │ │ └─BatchNorm2d: 5-40 128 │ │ │ │ └─ReLU: 5-41 -- │ │ │ └─Attention: 4-61 -- │ │ │ │ └─Identity: 5-42 -- │ │ │ └─Conv2dReLU: 4-62 -- │ │ │ │ └─Conv2d: 5-43 36,864 │ │ │ │ └─BatchNorm2d: 5-44 128 │ │ │ │ └─ReLU: 5-45 -- │ │ │ └─Attention: 4-63 -- │ │ │ │ └─Identity: 5-46 -- │ │ └─DecoderBlock: 3-14 -- │ │ │ └─Conv2dReLU: 4-64 -- │ │ │ │ └─Conv2d: 5-47 110,592 │ │ │ │ └─BatchNorm2d: 5-48 128 │ │ │ │ └─ReLU: 5-49 -- │ │ │ └─Attention: 4-65 -- │ │ │ │ └─Identity: 5-50 -- │ │ │ └─Conv2dReLU: 4-66 -- │ │ │ │ └─Conv2d: 5-51 36,864 │ │ │ │ └─BatchNorm2d: 5-52 128 │ │ │ │ └─ReLU: 5-53 -- │ │ │ └─Attention: 4-67 -- │ │ │ │ └─Identity: 5-54 -- │ │ └─DecoderBlock: 3-15 -- │ │ │ └─Conv2dReLU: 4-68 -- │ │ │ │ └─Conv2d: 5-55 92,160 │ │ │ │ └─BatchNorm2d: 5-56 64 │ │ │ │ └─ReLU: 5-57 -- │ │ │ └─Attention: 4-69 -- │ │ │ │ └─Identity: 5-58 -- │ │ │ └─Conv2dReLU: 4-70 -- │ │ │ │ └─Conv2d: 5-59 9,216 │ │ │ │ └─BatchNorm2d: 5-60 64 │ │ │ │ └─ReLU: 5-61 -- │ │ │ └─Attention: 4-71 -- │ │ │ │ └─Identity: 5-62 -- │ │ └─DecoderBlock: 3-16 -- │ │ │ └─Conv2dReLU: 4-72 -- │ │ │ │ └─Conv2d: 5-63 147,456 │ │ │ │ └─BatchNorm2d: 5-64 128 │ │ │ │ └─ReLU: 5-65 -- │ │ │ └─Attention: 4-73 -- │ │ │ │ └─Identity: 5-66 -- │ │ │ └─Conv2dReLU: 4-74 -- │ │ │ │ └─Conv2d: 5-67 36,864 │ │ │ │ └─BatchNorm2d: 5-68 128 │ │ │ │ └─ReLU: 5-69 -- │ │ │ └─Attention: 4-75 -- │ │ │ │ └─Identity: 5-70 -- │ │ └─DecoderBlock: 3-17 -- │ │ │ └─Conv2dReLU: 4-76 -- │ │ │ │ └─Conv2d: 5-71 110,592 │ │ │ │ └─BatchNorm2d: 5-72 128 │ │ │ │ └─ReLU: 5-73 -- │ │ │ └─Attention: 4-77 -- │ │ │ │ └─Identity: 5-74 -- │ │ │ └─Conv2dReLU: 4-78 -- │ │ │ │ └─Conv2d: 5-75 36,864 │ │ │ │ └─BatchNorm2d: 5-76 128 │ │ │ │ └─ReLU: 5-77 -- │ │ │ └─Attention: 4-79 -- │ │ │ │ └─Identity: 5-78 -- │ │ └─DecoderBlock: 3-18 -- │ │ │ └─Conv2dReLU: 4-80 -- │ │ │ │ └─Conv2d: 5-79 73,728 │ │ │ │ └─BatchNorm2d: 5-80 128 │ │ │ │ └─ReLU: 5-81 -- │ │ │ └─Attention: 4-81 -- │ │ │ │ └─Identity: 5-82 -- │ │ │ └─Conv2dReLU: 4-82 -- │ │ │ │ └─Conv2d: 5-83 36,864 │ │ │ │ └─BatchNorm2d: 5-84 128 │ │ │ │ └─ReLU: 5-85 -- │ │ │ └─Attention: 4-83 -- │ │ │ │ └─Identity: 5-86 -- │ │ └─DecoderBlock: 3-19 -- │ │ │ └─Conv2dReLU: 4-84 -- │ │ │ │ └─Conv2d: 5-87 4,608 │ │ │ │ └─BatchNorm2d: 5-88 32 │ │ │ │ └─ReLU: 5-89 -- │ │ │ └─Attention: 4-85 -- │ │ │ │ └─Identity: 5-90 -- │ │ │ └─Conv2dReLU: 4-86 -- │ │ │ │ └─Conv2d: 5-91 2,304 │ │ │ │ └─BatchNorm2d: 5-92 32 │ │ │ │ └─ReLU: 5-93 -- │ │ │ └─Attention: 4-87 -- │ │ │ │ └─Identity: 5-94 -- ├─SegmentationHead: 1-3 -- │ └─Conv2d: 2-11 290 │ └─Identity: 2-12 -- │ └─Activation: 2-13 -- │ │ └─Identity: 3-20 -- ====================================================================== Total params: 15,964,322 Trainable params: 15,964,322 Non-trainable params: 0 ======================================================================