Downloads: 1
Diff of
/model-description/architecture.txt
[000000]
..
[637b40]
Switch to unified view
| a | b/model-description/architecture.txt | ||
|---|---|---|---|
| 1 | ==================================================================================================== |
||
| 2 | Layer (type:depth-idx) Input Shape Output Shape |
||
| 3 | ==================================================================================================== |
||
| 4 | Unet -- -- |
||
| 5 | ├─UnetDecoder: 1 -- -- |
||
| 6 | │ └─ModuleList: 2-1 -- -- |
||
| 7 | ├─EfficientNetEncoder: 1-1 [1, 3, 512, 512] [1, 3, 512, 512] |
||
| 8 | │ └─Conv2d: 2-2 [1, 3, 512, 512] [1, 48, 256, 256] |
||
| 9 | │ └─BatchNorm2d: 2-3 [1, 48, 256, 256] [1, 48, 256, 256] |
||
| 10 | │ └─Swish: 2-4 [1, 48, 256, 256] [1, 48, 256, 256] |
||
| 11 | │ └─Sequential: 2 -- -- |
||
| 12 | │ │ └─Sequential: 3-1 [1, 48, 256, 256] [1, 24, 256, 256] |
||
| 13 | │ │ │ └─DepthwiseSeparableConv: 4-1 [1, 48, 256, 256] [1, 24, 256, 256] |
||
| 14 | │ │ │ │ └─Conv2d: 5-1 [1, 48, 256, 256] [1, 48, 256, 256] |
||
| 15 | │ │ │ │ └─BatchNorm2d: 5-2 [1, 48, 256, 256] [1, 48, 256, 256] |
||
| 16 | │ │ │ │ └─Swish: 5-3 [1, 48, 256, 256] [1, 48, 256, 256] |
||
| 17 | │ │ │ │ └─SqueezeExcite: 5-4 [1, 48, 256, 256] [1, 48, 256, 256] |
||
| 18 | │ │ │ │ └─Conv2d: 5-5 [1, 48, 256, 256] [1, 24, 256, 256] |
||
| 19 | │ │ │ │ └─BatchNorm2d: 5-6 [1, 24, 256, 256] [1, 24, 256, 256] |
||
| 20 | │ │ │ │ └─Identity: 5-7 [1, 24, 256, 256] [1, 24, 256, 256] |
||
| 21 | │ │ │ └─DepthwiseSeparableConv: 4-2 [1, 24, 256, 256] [1, 24, 256, 256] |
||
| 22 | │ │ │ │ └─Conv2d: 5-8 [1, 24, 256, 256] [1, 24, 256, 256] |
||
| 23 | │ │ │ │ └─BatchNorm2d: 5-9 [1, 24, 256, 256] [1, 24, 256, 256] |
||
| 24 | │ │ │ │ └─Swish: 5-10 [1, 24, 256, 256] [1, 24, 256, 256] |
||
| 25 | │ │ │ │ └─SqueezeExcite: 5-11 [1, 24, 256, 256] [1, 24, 256, 256] |
||
| 26 | │ │ │ │ └─Conv2d: 5-12 [1, 24, 256, 256] [1, 24, 256, 256] |
||
| 27 | │ │ │ │ └─BatchNorm2d: 5-13 [1, 24, 256, 256] [1, 24, 256, 256] |
||
| 28 | │ │ │ │ └─Identity: 5-14 [1, 24, 256, 256] [1, 24, 256, 256] |
||
| 29 | │ │ │ └─DepthwiseSeparableConv: 4-3 [1, 24, 256, 256] [1, 24, 256, 256] |
||
| 30 | │ │ │ │ └─Conv2d: 5-15 [1, 24, 256, 256] [1, 24, 256, 256] |
||
| 31 | │ │ │ │ └─BatchNorm2d: 5-16 [1, 24, 256, 256] [1, 24, 256, 256] |
||
| 32 | │ │ │ │ └─Swish: 5-17 [1, 24, 256, 256] [1, 24, 256, 256] |
||
| 33 | │ │ │ │ └─SqueezeExcite: 5-18 [1, 24, 256, 256] [1, 24, 256, 256] |
||
| 34 | │ │ │ │ └─Conv2d: 5-19 [1, 24, 256, 256] [1, 24, 256, 256] |
||
| 35 | │ │ │ │ └─BatchNorm2d: 5-20 [1, 24, 256, 256] [1, 24, 256, 256] |
||
| 36 | │ │ │ │ └─Identity: 5-21 [1, 24, 256, 256] [1, 24, 256, 256] |
||
| 37 | │ │ └─Sequential: 3-2 [1, 24, 256, 256] [1, 40, 128, 128] |
||
| 38 | │ │ │ └─InvertedResidual: 4-4 [1, 24, 256, 256] [1, 40, 128, 128] |
||
| 39 | │ │ │ │ └─Conv2d: 5-22 [1, 24, 256, 256] [1, 144, 256, 256] |
||
| 40 | │ │ │ │ └─BatchNorm2d: 5-23 [1, 144, 256, 256] [1, 144, 256, 256] |
||
| 41 | │ │ │ │ └─Swish: 5-24 [1, 144, 256, 256] [1, 144, 256, 256] |
||
| 42 | │ │ │ │ └─Conv2d: 5-25 [1, 144, 256, 256] [1, 144, 128, 128] |
||
| 43 | │ │ │ │ └─BatchNorm2d: 5-26 [1, 144, 128, 128] [1, 144, 128, 128] |
||
| 44 | │ │ │ │ └─Swish: 5-27 [1, 144, 128, 128] [1, 144, 128, 128] |
||
| 45 | │ │ │ │ └─SqueezeExcite: 5-28 [1, 144, 128, 128] [1, 144, 128, 128] |
||
| 46 | │ │ │ │ └─Conv2d: 5-29 [1, 144, 128, 128] [1, 40, 128, 128] |
||
| 47 | │ │ │ │ └─BatchNorm2d: 5-30 [1, 40, 128, 128] [1, 40, 128, 128] |
||
| 48 | │ │ │ └─InvertedResidual: 4-5 [1, 40, 128, 128] [1, 40, 128, 128] |
||
| 49 | │ │ │ │ └─Conv2d: 5-31 [1, 40, 128, 128] [1, 240, 128, 128] |
||
| 50 | │ │ │ │ └─BatchNorm2d: 5-32 [1, 240, 128, 128] [1, 240, 128, 128] |
||
| 51 | │ │ │ │ └─Swish: 5-33 [1, 240, 128, 128] [1, 240, 128, 128] |
||
| 52 | │ │ │ │ └─Conv2d: 5-34 [1, 240, 128, 128] [1, 240, 128, 128] |
||
| 53 | │ │ │ │ └─BatchNorm2d: 5-35 [1, 240, 128, 128] [1, 240, 128, 128] |
||
| 54 | │ │ │ │ └─Swish: 5-36 [1, 240, 128, 128] [1, 240, 128, 128] |
||
| 55 | │ │ │ │ └─SqueezeExcite: 5-37 [1, 240, 128, 128] [1, 240, 128, 128] |
||
| 56 | │ │ │ │ └─Conv2d: 5-38 [1, 240, 128, 128] [1, 40, 128, 128] |
||
| 57 | │ │ │ │ └─BatchNorm2d: 5-39 [1, 40, 128, 128] [1, 40, 128, 128] |
||
| 58 | │ │ │ └─InvertedResidual: 4-6 [1, 40, 128, 128] [1, 40, 128, 128] |
||
| 59 | │ │ │ │ └─Conv2d: 5-40 [1, 40, 128, 128] [1, 240, 128, 128] |
||
| 60 | │ │ │ │ └─BatchNorm2d: 5-41 [1, 240, 128, 128] [1, 240, 128, 128] |
||
| 61 | │ │ │ │ └─Swish: 5-42 [1, 240, 128, 128] [1, 240, 128, 128] |
||
| 62 | │ │ │ │ └─Conv2d: 5-43 [1, 240, 128, 128] [1, 240, 128, 128] |
||
| 63 | │ │ │ │ └─BatchNorm2d: 5-44 [1, 240, 128, 128] [1, 240, 128, 128] |
||
| 64 | │ │ │ │ └─Swish: 5-45 [1, 240, 128, 128] [1, 240, 128, 128] |
||
| 65 | │ │ │ │ └─SqueezeExcite: 5-46 [1, 240, 128, 128] [1, 240, 128, 128] |
||
| 66 | │ │ │ │ └─Conv2d: 5-47 [1, 240, 128, 128] [1, 40, 128, 128] |
||
| 67 | │ │ │ │ └─BatchNorm2d: 5-48 [1, 40, 128, 128] [1, 40, 128, 128] |
||
| 68 | │ │ │ └─InvertedResidual: 4-7 [1, 40, 128, 128] [1, 40, 128, 128] |
||
| 69 | │ │ │ │ └─Conv2d: 5-49 [1, 40, 128, 128] [1, 240, 128, 128] |
||
| 70 | │ │ │ │ └─BatchNorm2d: 5-50 [1, 240, 128, 128] [1, 240, 128, 128] |
||
| 71 | │ │ │ │ └─Swish: 5-51 [1, 240, 128, 128] [1, 240, 128, 128] |
||
| 72 | │ │ │ │ └─Conv2d: 5-52 [1, 240, 128, 128] [1, 240, 128, 128] |
||
| 73 | │ │ │ │ └─BatchNorm2d: 5-53 [1, 240, 128, 128] [1, 240, 128, 128] |
||
| 74 | │ │ │ │ └─Swish: 5-54 [1, 240, 128, 128] [1, 240, 128, 128] |
||
| 75 | │ │ │ │ └─SqueezeExcite: 5-55 [1, 240, 128, 128] [1, 240, 128, 128] |
||
| 76 | │ │ │ │ └─Conv2d: 5-56 [1, 240, 128, 128] [1, 40, 128, 128] |
||
| 77 | │ │ │ │ └─BatchNorm2d: 5-57 [1, 40, 128, 128] [1, 40, 128, 128] |
||
| 78 | │ │ │ └─InvertedResidual: 4-8 [1, 40, 128, 128] [1, 40, 128, 128] |
||
| 79 | │ │ │ │ └─Conv2d: 5-58 [1, 40, 128, 128] [1, 240, 128, 128] |
||
| 80 | │ │ │ │ └─BatchNorm2d: 5-59 [1, 240, 128, 128] [1, 240, 128, 128] |
||
| 81 | │ │ │ │ └─Swish: 5-60 [1, 240, 128, 128] [1, 240, 128, 128] |
||
| 82 | │ │ │ │ └─Conv2d: 5-61 [1, 240, 128, 128] [1, 240, 128, 128] |
||
| 83 | │ │ │ │ └─BatchNorm2d: 5-62 [1, 240, 128, 128] [1, 240, 128, 128] |
||
| 84 | │ │ │ │ └─Swish: 5-63 [1, 240, 128, 128] [1, 240, 128, 128] |
||
| 85 | │ │ │ │ └─SqueezeExcite: 5-64 [1, 240, 128, 128] [1, 240, 128, 128] |
||
| 86 | │ │ │ │ └─Conv2d: 5-65 [1, 240, 128, 128] [1, 40, 128, 128] |
||
| 87 | │ │ │ │ └─BatchNorm2d: 5-66 [1, 40, 128, 128] [1, 40, 128, 128] |
||
| 88 | │ │ └─Sequential: 3-3 [1, 40, 128, 128] [1, 64, 64, 64] |
||
| 89 | │ │ │ └─InvertedResidual: 4-9 [1, 40, 128, 128] [1, 64, 64, 64] |
||
| 90 | │ │ │ │ └─Conv2d: 5-67 [1, 40, 128, 128] [1, 240, 128, 128] |
||
| 91 | │ │ │ │ └─BatchNorm2d: 5-68 [1, 240, 128, 128] [1, 240, 128, 128] |
||
| 92 | │ │ │ │ └─Swish: 5-69 [1, 240, 128, 128] [1, 240, 128, 128] |
||
| 93 | │ │ │ │ └─Conv2d: 5-70 [1, 240, 128, 128] [1, 240, 64, 64] |
||
| 94 | │ │ │ │ └─BatchNorm2d: 5-71 [1, 240, 64, 64] [1, 240, 64, 64] |
||
| 95 | │ │ │ │ └─Swish: 5-72 [1, 240, 64, 64] [1, 240, 64, 64] |
||
| 96 | │ │ │ │ └─SqueezeExcite: 5-73 [1, 240, 64, 64] [1, 240, 64, 64] |
||
| 97 | │ │ │ │ └─Conv2d: 5-74 [1, 240, 64, 64] [1, 64, 64, 64] |
||
| 98 | │ │ │ │ └─BatchNorm2d: 5-75 [1, 64, 64, 64] [1, 64, 64, 64] |
||
| 99 | │ │ │ └─InvertedResidual: 4-10 [1, 64, 64, 64] [1, 64, 64, 64] |
||
| 100 | │ │ │ │ └─Conv2d: 5-76 [1, 64, 64, 64] [1, 384, 64, 64] |
||
| 101 | │ │ │ │ └─BatchNorm2d: 5-77 [1, 384, 64, 64] [1, 384, 64, 64] |
||
| 102 | │ │ │ │ └─Swish: 5-78 [1, 384, 64, 64] [1, 384, 64, 64] |
||
| 103 | │ │ │ │ └─Conv2d: 5-79 [1, 384, 64, 64] [1, 384, 64, 64] |
||
| 104 | │ │ │ │ └─BatchNorm2d: 5-80 [1, 384, 64, 64] [1, 384, 64, 64] |
||
| 105 | │ │ │ │ └─Swish: 5-81 [1, 384, 64, 64] [1, 384, 64, 64] |
||
| 106 | │ │ │ │ └─SqueezeExcite: 5-82 [1, 384, 64, 64] [1, 384, 64, 64] |
||
| 107 | │ │ │ │ └─Conv2d: 5-83 [1, 384, 64, 64] [1, 64, 64, 64] |
||
| 108 | │ │ │ │ └─BatchNorm2d: 5-84 [1, 64, 64, 64] [1, 64, 64, 64] |
||
| 109 | │ │ │ └─InvertedResidual: 4-11 [1, 64, 64, 64] [1, 64, 64, 64] |
||
| 110 | │ │ │ │ └─Conv2d: 5-85 [1, 64, 64, 64] [1, 384, 64, 64] |
||
| 111 | │ │ │ │ └─BatchNorm2d: 5-86 [1, 384, 64, 64] [1, 384, 64, 64] |
||
| 112 | │ │ │ │ └─Swish: 5-87 [1, 384, 64, 64] [1, 384, 64, 64] |
||
| 113 | │ │ │ │ └─Conv2d: 5-88 [1, 384, 64, 64] [1, 384, 64, 64] |
||
| 114 | │ │ │ │ └─BatchNorm2d: 5-89 [1, 384, 64, 64] [1, 384, 64, 64] |
||
| 115 | │ │ │ │ └─Swish: 5-90 [1, 384, 64, 64] [1, 384, 64, 64] |
||
| 116 | │ │ │ │ └─SqueezeExcite: 5-91 [1, 384, 64, 64] [1, 384, 64, 64] |
||
| 117 | │ │ │ │ └─Conv2d: 5-92 [1, 384, 64, 64] [1, 64, 64, 64] |
||
| 118 | │ │ │ │ └─BatchNorm2d: 5-93 [1, 64, 64, 64] [1, 64, 64, 64] |
||
| 119 | │ │ │ └─InvertedResidual: 4-12 [1, 64, 64, 64] [1, 64, 64, 64] |
||
| 120 | │ │ │ │ └─Conv2d: 5-94 [1, 64, 64, 64] [1, 384, 64, 64] |
||
| 121 | │ │ │ │ └─BatchNorm2d: 5-95 [1, 384, 64, 64] [1, 384, 64, 64] |
||
| 122 | │ │ │ │ └─Swish: 5-96 [1, 384, 64, 64] [1, 384, 64, 64] |
||
| 123 | │ │ │ │ └─Conv2d: 5-97 [1, 384, 64, 64] [1, 384, 64, 64] |
||
| 124 | │ │ │ │ └─BatchNorm2d: 5-98 [1, 384, 64, 64] [1, 384, 64, 64] |
||
| 125 | │ │ │ │ └─Swish: 5-99 [1, 384, 64, 64] [1, 384, 64, 64] |
||
| 126 | │ │ │ │ └─SqueezeExcite: 5-100 [1, 384, 64, 64] [1, 384, 64, 64] |
||
| 127 | │ │ │ │ └─Conv2d: 5-101 [1, 384, 64, 64] [1, 64, 64, 64] |
||
| 128 | │ │ │ │ └─BatchNorm2d: 5-102 [1, 64, 64, 64] [1, 64, 64, 64] |
||
| 129 | │ │ │ └─InvertedResidual: 4-13 [1, 64, 64, 64] [1, 64, 64, 64] |
||
| 130 | │ │ │ │ └─Conv2d: 5-103 [1, 64, 64, 64] [1, 384, 64, 64] |
||
| 131 | │ │ │ │ └─BatchNorm2d: 5-104 [1, 384, 64, 64] [1, 384, 64, 64] |
||
| 132 | │ │ │ │ └─Swish: 5-105 [1, 384, 64, 64] [1, 384, 64, 64] |
||
| 133 | │ │ │ │ └─Conv2d: 5-106 [1, 384, 64, 64] [1, 384, 64, 64] |
||
| 134 | │ │ │ │ └─BatchNorm2d: 5-107 [1, 384, 64, 64] [1, 384, 64, 64] |
||
| 135 | │ │ │ │ └─Swish: 5-108 [1, 384, 64, 64] [1, 384, 64, 64] |
||
| 136 | │ │ │ │ └─SqueezeExcite: 5-109 [1, 384, 64, 64] [1, 384, 64, 64] |
||
| 137 | │ │ │ │ └─Conv2d: 5-110 [1, 384, 64, 64] [1, 64, 64, 64] |
||
| 138 | │ │ │ │ └─BatchNorm2d: 5-111 [1, 64, 64, 64] [1, 64, 64, 64] |
||
| 139 | │ │ └─Sequential: 3-4 [1, 64, 64, 64] [1, 128, 32, 32] |
||
| 140 | │ │ │ └─InvertedResidual: 4-14 [1, 64, 64, 64] [1, 128, 32, 32] |
||
| 141 | │ │ │ │ └─Conv2d: 5-112 [1, 64, 64, 64] [1, 384, 64, 64] |
||
| 142 | │ │ │ │ └─BatchNorm2d: 5-113 [1, 384, 64, 64] [1, 384, 64, 64] |
||
| 143 | │ │ │ │ └─Swish: 5-114 [1, 384, 64, 64] [1, 384, 64, 64] |
||
| 144 | │ │ │ │ └─Conv2d: 5-115 [1, 384, 64, 64] [1, 384, 32, 32] |
||
| 145 | │ │ │ │ └─BatchNorm2d: 5-116 [1, 384, 32, 32] [1, 384, 32, 32] |
||
| 146 | │ │ │ │ └─Swish: 5-117 [1, 384, 32, 32] [1, 384, 32, 32] |
||
| 147 | │ │ │ │ └─SqueezeExcite: 5-118 [1, 384, 32, 32] [1, 384, 32, 32] |
||
| 148 | │ │ │ │ └─Conv2d: 5-119 [1, 384, 32, 32] [1, 128, 32, 32] |
||
| 149 | │ │ │ │ └─BatchNorm2d: 5-120 [1, 128, 32, 32] [1, 128, 32, 32] |
||
| 150 | │ │ │ └─InvertedResidual: 4-15 [1, 128, 32, 32] [1, 128, 32, 32] |
||
| 151 | │ │ │ │ └─Conv2d: 5-121 [1, 128, 32, 32] [1, 768, 32, 32] |
||
| 152 | │ │ │ │ └─BatchNorm2d: 5-122 [1, 768, 32, 32] [1, 768, 32, 32] |
||
| 153 | │ │ │ │ └─Swish: 5-123 [1, 768, 32, 32] [1, 768, 32, 32] |
||
| 154 | │ │ │ │ └─Conv2d: 5-124 [1, 768, 32, 32] [1, 768, 32, 32] |
||
| 155 | │ │ │ │ └─BatchNorm2d: 5-125 [1, 768, 32, 32] [1, 768, 32, 32] |
||
| 156 | │ │ │ │ └─Swish: 5-126 [1, 768, 32, 32] [1, 768, 32, 32] |
||
| 157 | │ │ │ │ └─SqueezeExcite: 5-127 [1, 768, 32, 32] [1, 768, 32, 32] |
||
| 158 | │ │ │ │ └─Conv2d: 5-128 [1, 768, 32, 32] [1, 128, 32, 32] |
||
| 159 | │ │ │ │ └─BatchNorm2d: 5-129 [1, 128, 32, 32] [1, 128, 32, 32] |
||
| 160 | │ │ │ └─InvertedResidual: 4-16 [1, 128, 32, 32] [1, 128, 32, 32] |
||
| 161 | │ │ │ │ └─Conv2d: 5-130 [1, 128, 32, 32] [1, 768, 32, 32] |
||
| 162 | │ │ │ │ └─BatchNorm2d: 5-131 [1, 768, 32, 32] [1, 768, 32, 32] |
||
| 163 | │ │ │ │ └─Swish: 5-132 [1, 768, 32, 32] [1, 768, 32, 32] |
||
| 164 | │ │ │ │ └─Conv2d: 5-133 [1, 768, 32, 32] [1, 768, 32, 32] |
||
| 165 | │ │ │ │ └─BatchNorm2d: 5-134 [1, 768, 32, 32] [1, 768, 32, 32] |
||
| 166 | │ │ │ │ └─Swish: 5-135 [1, 768, 32, 32] [1, 768, 32, 32] |
||
| 167 | │ │ │ │ └─SqueezeExcite: 5-136 [1, 768, 32, 32] [1, 768, 32, 32] |
||
| 168 | │ │ │ │ └─Conv2d: 5-137 [1, 768, 32, 32] [1, 128, 32, 32] |
||
| 169 | │ │ │ │ └─BatchNorm2d: 5-138 [1, 128, 32, 32] [1, 128, 32, 32] |
||
| 170 | │ │ │ └─InvertedResidual: 4-17 [1, 128, 32, 32] [1, 128, 32, 32] |
||
| 171 | │ │ │ │ └─Conv2d: 5-139 [1, 128, 32, 32] [1, 768, 32, 32] |
||
| 172 | │ │ │ │ └─BatchNorm2d: 5-140 [1, 768, 32, 32] [1, 768, 32, 32] |
||
| 173 | │ │ │ │ └─Swish: 5-141 [1, 768, 32, 32] [1, 768, 32, 32] |
||
| 174 | │ │ │ │ └─Conv2d: 5-142 [1, 768, 32, 32] [1, 768, 32, 32] |
||
| 175 | │ │ │ │ └─BatchNorm2d: 5-143 [1, 768, 32, 32] [1, 768, 32, 32] |
||
| 176 | │ │ │ │ └─Swish: 5-144 [1, 768, 32, 32] [1, 768, 32, 32] |
||
| 177 | │ │ │ │ └─SqueezeExcite: 5-145 [1, 768, 32, 32] [1, 768, 32, 32] |
||
| 178 | │ │ │ │ └─Conv2d: 5-146 [1, 768, 32, 32] [1, 128, 32, 32] |
||
| 179 | │ │ │ │ └─BatchNorm2d: 5-147 [1, 128, 32, 32] [1, 128, 32, 32] |
||
| 180 | │ │ │ └─InvertedResidual: 4-18 [1, 128, 32, 32] [1, 128, 32, 32] |
||
| 181 | │ │ │ │ └─Conv2d: 5-148 [1, 128, 32, 32] [1, 768, 32, 32] |
||
| 182 | │ │ │ │ └─BatchNorm2d: 5-149 [1, 768, 32, 32] [1, 768, 32, 32] |
||
| 183 | │ │ │ │ └─Swish: 5-150 [1, 768, 32, 32] [1, 768, 32, 32] |
||
| 184 | │ │ │ │ └─Conv2d: 5-151 [1, 768, 32, 32] [1, 768, 32, 32] |
||
| 185 | │ │ │ │ └─BatchNorm2d: 5-152 [1, 768, 32, 32] [1, 768, 32, 32] |
||
| 186 | │ │ │ │ └─Swish: 5-153 [1, 768, 32, 32] [1, 768, 32, 32] |
||
| 187 | │ │ │ │ └─SqueezeExcite: 5-154 [1, 768, 32, 32] [1, 768, 32, 32] |
||
| 188 | │ │ │ │ └─Conv2d: 5-155 [1, 768, 32, 32] [1, 128, 32, 32] |
||
| 189 | │ │ │ │ └─BatchNorm2d: 5-156 [1, 128, 32, 32] [1, 128, 32, 32] |
||
| 190 | │ │ │ └─InvertedResidual: 4-19 [1, 128, 32, 32] [1, 128, 32, 32] |
||
| 191 | │ │ │ │ └─Conv2d: 5-157 [1, 128, 32, 32] [1, 768, 32, 32] |
||
| 192 | │ │ │ │ └─BatchNorm2d: 5-158 [1, 768, 32, 32] [1, 768, 32, 32] |
||
| 193 | │ │ │ │ └─Swish: 5-159 [1, 768, 32, 32] [1, 768, 32, 32] |
||
| 194 | │ │ │ │ └─Conv2d: 5-160 [1, 768, 32, 32] [1, 768, 32, 32] |
||
| 195 | │ │ │ │ └─BatchNorm2d: 5-161 [1, 768, 32, 32] [1, 768, 32, 32] |
||
| 196 | │ │ │ │ └─Swish: 5-162 [1, 768, 32, 32] [1, 768, 32, 32] |
||
| 197 | │ │ │ │ └─SqueezeExcite: 5-163 [1, 768, 32, 32] [1, 768, 32, 32] |
||
| 198 | │ │ │ │ └─Conv2d: 5-164 [1, 768, 32, 32] [1, 128, 32, 32] |
||
| 199 | │ │ │ │ └─BatchNorm2d: 5-165 [1, 128, 32, 32] [1, 128, 32, 32] |
||
| 200 | │ │ │ └─InvertedResidual: 4-20 [1, 128, 32, 32] [1, 128, 32, 32] |
||
| 201 | │ │ │ │ └─Conv2d: 5-166 [1, 128, 32, 32] [1, 768, 32, 32] |
||
| 202 | │ │ │ │ └─BatchNorm2d: 5-167 [1, 768, 32, 32] [1, 768, 32, 32] |
||
| 203 | │ │ │ │ └─Swish: 5-168 [1, 768, 32, 32] [1, 768, 32, 32] |
||
| 204 | │ │ │ │ └─Conv2d: 5-169 [1, 768, 32, 32] [1, 768, 32, 32] |
||
| 205 | │ │ │ │ └─BatchNorm2d: 5-170 [1, 768, 32, 32] [1, 768, 32, 32] |
||
| 206 | │ │ │ │ └─Swish: 5-171 [1, 768, 32, 32] [1, 768, 32, 32] |
||
| 207 | │ │ │ │ └─SqueezeExcite: 5-172 [1, 768, 32, 32] [1, 768, 32, 32] |
||
| 208 | │ │ │ │ └─Conv2d: 5-173 [1, 768, 32, 32] [1, 128, 32, 32] |
||
| 209 | │ │ │ │ └─BatchNorm2d: 5-174 [1, 128, 32, 32] [1, 128, 32, 32] |
||
| 210 | │ │ └─Sequential: 3-5 [1, 128, 32, 32] [1, 176, 32, 32] |
||
| 211 | │ │ │ └─InvertedResidual: 4-21 [1, 128, 32, 32] [1, 176, 32, 32] |
||
| 212 | │ │ │ │ └─Conv2d: 5-175 [1, 128, 32, 32] [1, 768, 32, 32] |
||
| 213 | │ │ │ │ └─BatchNorm2d: 5-176 [1, 768, 32, 32] [1, 768, 32, 32] |
||
| 214 | │ │ │ │ └─Swish: 5-177 [1, 768, 32, 32] [1, 768, 32, 32] |
||
| 215 | │ │ │ │ └─Conv2d: 5-178 [1, 768, 32, 32] [1, 768, 32, 32] |
||
| 216 | │ │ │ │ └─BatchNorm2d: 5-179 [1, 768, 32, 32] [1, 768, 32, 32] |
||
| 217 | │ │ │ │ └─Swish: 5-180 [1, 768, 32, 32] [1, 768, 32, 32] |
||
| 218 | │ │ │ │ └─SqueezeExcite: 5-181 [1, 768, 32, 32] [1, 768, 32, 32] |
||
| 219 | │ │ │ │ └─Conv2d: 5-182 [1, 768, 32, 32] [1, 176, 32, 32] |
||
| 220 | │ │ │ │ └─BatchNorm2d: 5-183 [1, 176, 32, 32] [1, 176, 32, 32] |
||
| 221 | │ │ │ └─InvertedResidual: 4-22 [1, 176, 32, 32] [1, 176, 32, 32] |
||
| 222 | │ │ │ │ └─Conv2d: 5-184 [1, 176, 32, 32] [1, 1056, 32, 32] |
||
| 223 | │ │ │ │ └─BatchNorm2d: 5-185 [1, 1056, 32, 32] [1, 1056, 32, 32] |
||
| 224 | │ │ │ │ └─Swish: 5-186 [1, 1056, 32, 32] [1, 1056, 32, 32] |
||
| 225 | │ │ │ │ └─Conv2d: 5-187 [1, 1056, 32, 32] [1, 1056, 32, 32] |
||
| 226 | │ │ │ │ └─BatchNorm2d: 5-188 [1, 1056, 32, 32] [1, 1056, 32, 32] |
||
| 227 | │ │ │ │ └─Swish: 5-189 [1, 1056, 32, 32] [1, 1056, 32, 32] |
||
| 228 | │ │ │ │ └─SqueezeExcite: 5-190 [1, 1056, 32, 32] [1, 1056, 32, 32] |
||
| 229 | │ │ │ │ └─Conv2d: 5-191 [1, 1056, 32, 32] [1, 176, 32, 32] |
||
| 230 | │ │ │ │ └─BatchNorm2d: 5-192 [1, 176, 32, 32] [1, 176, 32, 32] |
||
| 231 | │ │ │ └─InvertedResidual: 4-23 [1, 176, 32, 32] [1, 176, 32, 32] |
||
| 232 | │ │ │ │ └─Conv2d: 5-193 [1, 176, 32, 32] [1, 1056, 32, 32] |
||
| 233 | │ │ │ │ └─BatchNorm2d: 5-194 [1, 1056, 32, 32] [1, 1056, 32, 32] |
||
| 234 | │ │ │ │ └─Swish: 5-195 [1, 1056, 32, 32] [1, 1056, 32, 32] |
||
| 235 | │ │ │ │ └─Conv2d: 5-196 [1, 1056, 32, 32] [1, 1056, 32, 32] |
||
| 236 | │ │ │ │ └─BatchNorm2d: 5-197 [1, 1056, 32, 32] [1, 1056, 32, 32] |
||
| 237 | │ │ │ │ └─Swish: 5-198 [1, 1056, 32, 32] [1, 1056, 32, 32] |
||
| 238 | │ │ │ │ └─SqueezeExcite: 5-199 [1, 1056, 32, 32] [1, 1056, 32, 32] |
||
| 239 | │ │ │ │ └─Conv2d: 5-200 [1, 1056, 32, 32] [1, 176, 32, 32] |
||
| 240 | │ │ │ │ └─BatchNorm2d: 5-201 [1, 176, 32, 32] [1, 176, 32, 32] |
||
| 241 | │ │ │ └─InvertedResidual: 4-24 [1, 176, 32, 32] [1, 176, 32, 32] |
||
| 242 | │ │ │ │ └─Conv2d: 5-202 [1, 176, 32, 32] [1, 1056, 32, 32] |
||
| 243 | │ │ │ │ └─BatchNorm2d: 5-203 [1, 1056, 32, 32] [1, 1056, 32, 32] |
||
| 244 | │ │ │ │ └─Swish: 5-204 [1, 1056, 32, 32] [1, 1056, 32, 32] |
||
| 245 | │ │ │ │ └─Conv2d: 5-205 [1, 1056, 32, 32] [1, 1056, 32, 32] |
||
| 246 | │ │ │ │ └─BatchNorm2d: 5-206 [1, 1056, 32, 32] [1, 1056, 32, 32] |
||
| 247 | │ │ │ │ └─Swish: 5-207 [1, 1056, 32, 32] [1, 1056, 32, 32] |
||
| 248 | │ │ │ │ └─SqueezeExcite: 5-208 [1, 1056, 32, 32] [1, 1056, 32, 32] |
||
| 249 | │ │ │ │ └─Conv2d: 5-209 [1, 1056, 32, 32] [1, 176, 32, 32] |
||
| 250 | │ │ │ │ └─BatchNorm2d: 5-210 [1, 176, 32, 32] [1, 176, 32, 32] |
||
| 251 | │ │ │ └─InvertedResidual: 4-25 [1, 176, 32, 32] [1, 176, 32, 32] |
||
| 252 | │ │ │ │ └─Conv2d: 5-211 [1, 176, 32, 32] [1, 1056, 32, 32] |
||
| 253 | │ │ │ │ └─BatchNorm2d: 5-212 [1, 1056, 32, 32] [1, 1056, 32, 32] |
||
| 254 | │ │ │ │ └─Swish: 5-213 [1, 1056, 32, 32] [1, 1056, 32, 32] |
||
| 255 | │ │ │ │ └─Conv2d: 5-214 [1, 1056, 32, 32] [1, 1056, 32, 32] |
||
| 256 | │ │ │ │ └─BatchNorm2d: 5-215 [1, 1056, 32, 32] [1, 1056, 32, 32] |
||
| 257 | │ │ │ │ └─Swish: 5-216 [1, 1056, 32, 32] [1, 1056, 32, 32] |
||
| 258 | │ │ │ │ └─SqueezeExcite: 5-217 [1, 1056, 32, 32] [1, 1056, 32, 32] |
||
| 259 | │ │ │ │ └─Conv2d: 5-218 [1, 1056, 32, 32] [1, 176, 32, 32] |
||
| 260 | │ │ │ │ └─BatchNorm2d: 5-219 [1, 176, 32, 32] [1, 176, 32, 32] |
||
| 261 | │ │ │ └─InvertedResidual: 4-26 [1, 176, 32, 32] [1, 176, 32, 32] |
||
| 262 | │ │ │ │ └─Conv2d: 5-220 [1, 176, 32, 32] [1, 1056, 32, 32] |
||
| 263 | │ │ │ │ └─BatchNorm2d: 5-221 [1, 1056, 32, 32] [1, 1056, 32, 32] |
||
| 264 | │ │ │ │ └─Swish: 5-222 [1, 1056, 32, 32] [1, 1056, 32, 32] |
||
| 265 | │ │ │ │ └─Conv2d: 5-223 [1, 1056, 32, 32] [1, 1056, 32, 32] |
||
| 266 | │ │ │ │ └─BatchNorm2d: 5-224 [1, 1056, 32, 32] [1, 1056, 32, 32] |
||
| 267 | │ │ │ │ └─Swish: 5-225 [1, 1056, 32, 32] [1, 1056, 32, 32] |
||
| 268 | │ │ │ │ └─SqueezeExcite: 5-226 [1, 1056, 32, 32] [1, 1056, 32, 32] |
||
| 269 | │ │ │ │ └─Conv2d: 5-227 [1, 1056, 32, 32] [1, 176, 32, 32] |
||
| 270 | │ │ │ │ └─BatchNorm2d: 5-228 [1, 176, 32, 32] [1, 176, 32, 32] |
||
| 271 | │ │ │ └─InvertedResidual: 4-27 [1, 176, 32, 32] [1, 176, 32, 32] |
||
| 272 | │ │ │ │ └─Conv2d: 5-229 [1, 176, 32, 32] [1, 1056, 32, 32] |
||
| 273 | │ │ │ │ └─BatchNorm2d: 5-230 [1, 1056, 32, 32] [1, 1056, 32, 32] |
||
| 274 | │ │ │ │ └─Swish: 5-231 [1, 1056, 32, 32] [1, 1056, 32, 32] |
||
| 275 | │ │ │ │ └─Conv2d: 5-232 [1, 1056, 32, 32] [1, 1056, 32, 32] |
||
| 276 | │ │ │ │ └─BatchNorm2d: 5-233 [1, 1056, 32, 32] [1, 1056, 32, 32] |
||
| 277 | │ │ │ │ └─Swish: 5-234 [1, 1056, 32, 32] [1, 1056, 32, 32] |
||
| 278 | │ │ │ │ └─SqueezeExcite: 5-235 [1, 1056, 32, 32] [1, 1056, 32, 32] |
||
| 279 | │ │ │ │ └─Conv2d: 5-236 [1, 1056, 32, 32] [1, 176, 32, 32] |
||
| 280 | │ │ │ │ └─BatchNorm2d: 5-237 [1, 176, 32, 32] [1, 176, 32, 32] |
||
| 281 | │ │ └─Sequential: 3-6 [1, 176, 32, 32] [1, 304, 16, 16] |
||
| 282 | │ │ │ └─InvertedResidual: 4-28 [1, 176, 32, 32] [1, 304, 16, 16] |
||
| 283 | │ │ │ │ └─Conv2d: 5-238 [1, 176, 32, 32] [1, 1056, 32, 32] |
||
| 284 | │ │ │ │ └─BatchNorm2d: 5-239 [1, 1056, 32, 32] [1, 1056, 32, 32] |
||
| 285 | │ │ │ │ └─Swish: 5-240 [1, 1056, 32, 32] [1, 1056, 32, 32] |
||
| 286 | │ │ │ │ └─Conv2d: 5-241 [1, 1056, 32, 32] [1, 1056, 16, 16] |
||
| 287 | │ │ │ │ └─BatchNorm2d: 5-242 [1, 1056, 16, 16] [1, 1056, 16, 16] |
||
| 288 | │ │ │ │ └─Swish: 5-243 [1, 1056, 16, 16] [1, 1056, 16, 16] |
||
| 289 | │ │ │ │ └─SqueezeExcite: 5-244 [1, 1056, 16, 16] [1, 1056, 16, 16] |
||
| 290 | │ │ │ │ └─Conv2d: 5-245 [1, 1056, 16, 16] [1, 304, 16, 16] |
||
| 291 | │ │ │ │ └─BatchNorm2d: 5-246 [1, 304, 16, 16] [1, 304, 16, 16] |
||
| 292 | │ │ │ └─InvertedResidual: 4-29 [1, 304, 16, 16] [1, 304, 16, 16] |
||
| 293 | │ │ │ │ └─Conv2d: 5-247 [1, 304, 16, 16] [1, 1824, 16, 16] |
||
| 294 | │ │ │ │ └─BatchNorm2d: 5-248 [1, 1824, 16, 16] [1, 1824, 16, 16] |
||
| 295 | │ │ │ │ └─Swish: 5-249 [1, 1824, 16, 16] [1, 1824, 16, 16] |
||
| 296 | │ │ │ │ └─Conv2d: 5-250 [1, 1824, 16, 16] [1, 1824, 16, 16] |
||
| 297 | │ │ │ │ └─BatchNorm2d: 5-251 [1, 1824, 16, 16] [1, 1824, 16, 16] |
||
| 298 | │ │ │ │ └─Swish: 5-252 [1, 1824, 16, 16] [1, 1824, 16, 16] |
||
| 299 | │ │ │ │ └─SqueezeExcite: 5-253 [1, 1824, 16, 16] [1, 1824, 16, 16] |
||
| 300 | │ │ │ │ └─Conv2d: 5-254 [1, 1824, 16, 16] [1, 304, 16, 16] |
||
| 301 | │ │ │ │ └─BatchNorm2d: 5-255 [1, 304, 16, 16] [1, 304, 16, 16] |
||
| 302 | │ │ │ └─InvertedResidual: 4-30 [1, 304, 16, 16] [1, 304, 16, 16] |
||
| 303 | │ │ │ │ └─Conv2d: 5-256 [1, 304, 16, 16] [1, 1824, 16, 16] |
||
| 304 | │ │ │ │ └─BatchNorm2d: 5-257 [1, 1824, 16, 16] [1, 1824, 16, 16] |
||
| 305 | │ │ │ │ └─Swish: 5-258 [1, 1824, 16, 16] [1, 1824, 16, 16] |
||
| 306 | │ │ │ │ └─Conv2d: 5-259 [1, 1824, 16, 16] [1, 1824, 16, 16] |
||
| 307 | │ │ │ │ └─BatchNorm2d: 5-260 [1, 1824, 16, 16] [1, 1824, 16, 16] |
||
| 308 | │ │ │ │ └─Swish: 5-261 [1, 1824, 16, 16] [1, 1824, 16, 16] |
||
| 309 | │ │ │ │ └─SqueezeExcite: 5-262 [1, 1824, 16, 16] [1, 1824, 16, 16] |
||
| 310 | │ │ │ │ └─Conv2d: 5-263 [1, 1824, 16, 16] [1, 304, 16, 16] |
||
| 311 | │ │ │ │ └─BatchNorm2d: 5-264 [1, 304, 16, 16] [1, 304, 16, 16] |
||
| 312 | │ │ │ └─InvertedResidual: 4-31 [1, 304, 16, 16] [1, 304, 16, 16] |
||
| 313 | │ │ │ │ └─Conv2d: 5-265 [1, 304, 16, 16] [1, 1824, 16, 16] |
||
| 314 | │ │ │ │ └─BatchNorm2d: 5-266 [1, 1824, 16, 16] [1, 1824, 16, 16] |
||
| 315 | │ │ │ │ └─Swish: 5-267 [1, 1824, 16, 16] [1, 1824, 16, 16] |
||
| 316 | │ │ │ │ └─Conv2d: 5-268 [1, 1824, 16, 16] [1, 1824, 16, 16] |
||
| 317 | │ │ │ │ └─BatchNorm2d: 5-269 [1, 1824, 16, 16] [1, 1824, 16, 16] |
||
| 318 | │ │ │ │ └─Swish: 5-270 [1, 1824, 16, 16] [1, 1824, 16, 16] |
||
| 319 | │ │ │ │ └─SqueezeExcite: 5-271 [1, 1824, 16, 16] [1, 1824, 16, 16] |
||
| 320 | │ │ │ │ └─Conv2d: 5-272 [1, 1824, 16, 16] [1, 304, 16, 16] |
||
| 321 | │ │ │ │ └─BatchNorm2d: 5-273 [1, 304, 16, 16] [1, 304, 16, 16] |
||
| 322 | │ │ │ └─InvertedResidual: 4-32 [1, 304, 16, 16] [1, 304, 16, 16] |
||
| 323 | │ │ │ │ └─Conv2d: 5-274 [1, 304, 16, 16] [1, 1824, 16, 16] |
||
| 324 | │ │ │ │ └─BatchNorm2d: 5-275 [1, 1824, 16, 16] [1, 1824, 16, 16] |
||
| 325 | │ │ │ │ └─Swish: 5-276 [1, 1824, 16, 16] [1, 1824, 16, 16] |
||
| 326 | │ │ │ │ └─Conv2d: 5-277 [1, 1824, 16, 16] [1, 1824, 16, 16] |
||
| 327 | │ │ │ │ └─BatchNorm2d: 5-278 [1, 1824, 16, 16] [1, 1824, 16, 16] |
||
| 328 | │ │ │ │ └─Swish: 5-279 [1, 1824, 16, 16] [1, 1824, 16, 16] |
||
| 329 | │ │ │ │ └─SqueezeExcite: 5-280 [1, 1824, 16, 16] [1, 1824, 16, 16] |
||
| 330 | │ │ │ │ └─Conv2d: 5-281 [1, 1824, 16, 16] [1, 304, 16, 16] |
||
| 331 | │ │ │ │ └─BatchNorm2d: 5-282 [1, 304, 16, 16] [1, 304, 16, 16] |
||
| 332 | │ │ │ └─InvertedResidual: 4-33 [1, 304, 16, 16] [1, 304, 16, 16] |
||
| 333 | │ │ │ │ └─Conv2d: 5-283 [1, 304, 16, 16] [1, 1824, 16, 16] |
||
| 334 | │ │ │ │ └─BatchNorm2d: 5-284 [1, 1824, 16, 16] [1, 1824, 16, 16] |
||
| 335 | │ │ │ │ └─Swish: 5-285 [1, 1824, 16, 16] [1, 1824, 16, 16] |
||
| 336 | │ │ │ │ └─Conv2d: 5-286 [1, 1824, 16, 16] [1, 1824, 16, 16] |
||
| 337 | │ │ │ │ └─BatchNorm2d: 5-287 [1, 1824, 16, 16] [1, 1824, 16, 16] |
||
| 338 | │ │ │ │ └─Swish: 5-288 [1, 1824, 16, 16] [1, 1824, 16, 16] |
||
| 339 | │ │ │ │ └─SqueezeExcite: 5-289 [1, 1824, 16, 16] [1, 1824, 16, 16] |
||
| 340 | │ │ │ │ └─Conv2d: 5-290 [1, 1824, 16, 16] [1, 304, 16, 16] |
||
| 341 | │ │ │ │ └─BatchNorm2d: 5-291 [1, 304, 16, 16] [1, 304, 16, 16] |
||
| 342 | │ │ │ └─InvertedResidual: 4-34 [1, 304, 16, 16] [1, 304, 16, 16] |
||
| 343 | │ │ │ │ └─Conv2d: 5-292 [1, 304, 16, 16] [1, 1824, 16, 16] |
||
| 344 | │ │ │ │ └─BatchNorm2d: 5-293 [1, 1824, 16, 16] [1, 1824, 16, 16] |
||
| 345 | │ │ │ │ └─Swish: 5-294 [1, 1824, 16, 16] [1, 1824, 16, 16] |
||
| 346 | │ │ │ │ └─Conv2d: 5-295 [1, 1824, 16, 16] [1, 1824, 16, 16] |
||
| 347 | │ │ │ │ └─BatchNorm2d: 5-296 [1, 1824, 16, 16] [1, 1824, 16, 16] |
||
| 348 | │ │ │ │ └─Swish: 5-297 [1, 1824, 16, 16] [1, 1824, 16, 16] |
||
| 349 | │ │ │ │ └─SqueezeExcite: 5-298 [1, 1824, 16, 16] [1, 1824, 16, 16] |
||
| 350 | │ │ │ │ └─Conv2d: 5-299 [1, 1824, 16, 16] [1, 304, 16, 16] |
||
| 351 | │ │ │ │ └─BatchNorm2d: 5-300 [1, 304, 16, 16] [1, 304, 16, 16] |
||
| 352 | │ │ │ └─InvertedResidual: 4-35 [1, 304, 16, 16] [1, 304, 16, 16] |
||
| 353 | │ │ │ │ └─Conv2d: 5-301 [1, 304, 16, 16] [1, 1824, 16, 16] |
||
| 354 | │ │ │ │ └─BatchNorm2d: 5-302 [1, 1824, 16, 16] [1, 1824, 16, 16] |
||
| 355 | │ │ │ │ └─Swish: 5-303 [1, 1824, 16, 16] [1, 1824, 16, 16] |
||
| 356 | │ │ │ │ └─Conv2d: 5-304 [1, 1824, 16, 16] [1, 1824, 16, 16] |
||
| 357 | │ │ │ │ └─BatchNorm2d: 5-305 [1, 1824, 16, 16] [1, 1824, 16, 16] |
||
| 358 | │ │ │ │ └─Swish: 5-306 [1, 1824, 16, 16] [1, 1824, 16, 16] |
||
| 359 | │ │ │ │ └─SqueezeExcite: 5-307 [1, 1824, 16, 16] [1, 1824, 16, 16] |
||
| 360 | │ │ │ │ └─Conv2d: 5-308 [1, 1824, 16, 16] [1, 304, 16, 16] |
||
| 361 | │ │ │ │ └─BatchNorm2d: 5-309 [1, 304, 16, 16] [1, 304, 16, 16] |
||
| 362 | │ │ │ └─InvertedResidual: 4-36 [1, 304, 16, 16] [1, 304, 16, 16] |
||
| 363 | │ │ │ │ └─Conv2d: 5-310 [1, 304, 16, 16] [1, 1824, 16, 16] |
||
| 364 | │ │ │ │ └─BatchNorm2d: 5-311 [1, 1824, 16, 16] [1, 1824, 16, 16] |
||
| 365 | │ │ │ │ └─Swish: 5-312 [1, 1824, 16, 16] [1, 1824, 16, 16] |
||
| 366 | │ │ │ │ └─Conv2d: 5-313 [1, 1824, 16, 16] [1, 1824, 16, 16] |
||
| 367 | │ │ │ │ └─BatchNorm2d: 5-314 [1, 1824, 16, 16] [1, 1824, 16, 16] |
||
| 368 | │ │ │ │ └─Swish: 5-315 [1, 1824, 16, 16] [1, 1824, 16, 16] |
||
| 369 | │ │ │ │ └─SqueezeExcite: 5-316 [1, 1824, 16, 16] [1, 1824, 16, 16] |
||
| 370 | │ │ │ │ └─Conv2d: 5-317 [1, 1824, 16, 16] [1, 304, 16, 16] |
||
| 371 | │ │ │ │ └─BatchNorm2d: 5-318 [1, 304, 16, 16] [1, 304, 16, 16] |
||
| 372 | │ │ └─Sequential: 3-7 [1, 304, 16, 16] [1, 512, 16, 16] |
||
| 373 | │ │ │ └─InvertedResidual: 4-37 [1, 304, 16, 16] [1, 512, 16, 16] |
||
| 374 | │ │ │ │ └─Conv2d: 5-319 [1, 304, 16, 16] [1, 1824, 16, 16] |
||
| 375 | │ │ │ │ └─BatchNorm2d: 5-320 [1, 1824, 16, 16] [1, 1824, 16, 16] |
||
| 376 | │ │ │ │ └─Swish: 5-321 [1, 1824, 16, 16] [1, 1824, 16, 16] |
||
| 377 | │ │ │ │ └─Conv2d: 5-322 [1, 1824, 16, 16] [1, 1824, 16, 16] |
||
| 378 | │ │ │ │ └─BatchNorm2d: 5-323 [1, 1824, 16, 16] [1, 1824, 16, 16] |
||
| 379 | │ │ │ │ └─Swish: 5-324 [1, 1824, 16, 16] [1, 1824, 16, 16] |
||
| 380 | │ │ │ │ └─SqueezeExcite: 5-325 [1, 1824, 16, 16] [1, 1824, 16, 16] |
||
| 381 | │ │ │ │ └─Conv2d: 5-326 [1, 1824, 16, 16] [1, 512, 16, 16] |
||
| 382 | │ │ │ │ └─BatchNorm2d: 5-327 [1, 512, 16, 16] [1, 512, 16, 16] |
||
| 383 | │ │ │ └─InvertedResidual: 4-38 [1, 512, 16, 16] [1, 512, 16, 16] |
||
| 384 | │ │ │ │ └─Conv2d: 5-328 [1, 512, 16, 16] [1, 3072, 16, 16] |
||
| 385 | │ │ │ │ └─BatchNorm2d: 5-329 [1, 3072, 16, 16] [1, 3072, 16, 16] |
||
| 386 | │ │ │ │ └─Swish: 5-330 [1, 3072, 16, 16] [1, 3072, 16, 16] |
||
| 387 | │ │ │ │ └─Conv2d: 5-331 [1, 3072, 16, 16] [1, 3072, 16, 16] |
||
| 388 | │ │ │ │ └─BatchNorm2d: 5-332 [1, 3072, 16, 16] [1, 3072, 16, 16] |
||
| 389 | │ │ │ │ └─Swish: 5-333 [1, 3072, 16, 16] [1, 3072, 16, 16] |
||
| 390 | │ │ │ │ └─SqueezeExcite: 5-334 [1, 3072, 16, 16] [1, 3072, 16, 16] |
||
| 391 | │ │ │ │ └─Conv2d: 5-335 [1, 3072, 16, 16] [1, 512, 16, 16] |
||
| 392 | │ │ │ │ └─BatchNorm2d: 5-336 [1, 512, 16, 16] [1, 512, 16, 16] |
||
| 393 | │ │ │ └─InvertedResidual: 4-39 [1, 512, 16, 16] [1, 512, 16, 16] |
||
| 394 | │ │ │ │ └─Conv2d: 5-337 [1, 512, 16, 16] [1, 3072, 16, 16] |
||
| 395 | │ │ │ │ └─BatchNorm2d: 5-338 [1, 3072, 16, 16] [1, 3072, 16, 16] |
||
| 396 | │ │ │ │ └─Swish: 5-339 [1, 3072, 16, 16] [1, 3072, 16, 16] |
||
| 397 | │ │ │ │ └─Conv2d: 5-340 [1, 3072, 16, 16] [1, 3072, 16, 16] |
||
| 398 | │ │ │ │ └─BatchNorm2d: 5-341 [1, 3072, 16, 16] [1, 3072, 16, 16] |
||
| 399 | │ │ │ │ └─Swish: 5-342 [1, 3072, 16, 16] [1, 3072, 16, 16] |
||
| 400 | │ │ │ │ └─SqueezeExcite: 5-343 [1, 3072, 16, 16] [1, 3072, 16, 16] |
||
| 401 | │ │ │ │ └─Conv2d: 5-344 [1, 3072, 16, 16] [1, 512, 16, 16] |
||
| 402 | │ │ │ │ └─BatchNorm2d: 5-345 [1, 512, 16, 16] [1, 512, 16, 16] |
||
| 403 | ├─UnetDecoder: 1-2 [1, 3, 512, 512] [1, 32, 512, 512] |
||
| 404 | │ └─Identity: 2-5 [1, 512, 16, 16] [1, 512, 16, 16] |
||
| 405 | │ └─ModuleList: 2-1 -- -- |
||
| 406 | │ │ └─DecoderBlock: 3-8 [1, 512, 16, 16] [1, 512, 32, 32] |
||
| 407 | │ │ │ └─Attention: 4-40 [1, 688, 32, 32] [1, 688, 32, 32] |
||
| 408 | │ │ │ │ └─Identity: 5-346 [1, 688, 32, 32] [1, 688, 32, 32] |
||
| 409 | │ │ │ └─Conv2dReLU: 4-41 [1, 688, 32, 32] [1, 512, 32, 32] |
||
| 410 | │ │ │ │ └─Conv2d: 5-347 [1, 688, 32, 32] [1, 512, 32, 32] |
||
| 411 | │ │ │ │ └─BatchNorm2d: 5-348 [1, 512, 32, 32] [1, 512, 32, 32] |
||
| 412 | │ │ │ │ └─ReLU: 5-349 [1, 512, 32, 32] [1, 512, 32, 32] |
||
| 413 | │ │ │ └─Conv2dReLU: 4-42 [1, 512, 32, 32] [1, 512, 32, 32] |
||
| 414 | │ │ │ │ └─Conv2d: 5-350 [1, 512, 32, 32] [1, 512, 32, 32] |
||
| 415 | │ │ │ │ └─BatchNorm2d: 5-351 [1, 512, 32, 32] [1, 512, 32, 32] |
||
| 416 | │ │ │ │ └─ReLU: 5-352 [1, 512, 32, 32] [1, 512, 32, 32] |
||
| 417 | │ │ │ └─Attention: 4-43 [1, 512, 32, 32] [1, 512, 32, 32] |
||
| 418 | │ │ │ │ └─Identity: 5-353 [1, 512, 32, 32] [1, 512, 32, 32] |
||
| 419 | │ │ └─DecoderBlock: 3-9 [1, 512, 32, 32] [1, 256, 64, 64] |
||
| 420 | │ │ │ └─Attention: 4-44 [1, 576, 64, 64] [1, 576, 64, 64] |
||
| 421 | │ │ │ │ └─Identity: 5-354 [1, 576, 64, 64] [1, 576, 64, 64] |
||
| 422 | │ │ │ └─Conv2dReLU: 4-45 [1, 576, 64, 64] [1, 256, 64, 64] |
||
| 423 | │ │ │ │ └─Conv2d: 5-355 [1, 576, 64, 64] [1, 256, 64, 64] |
||
| 424 | │ │ │ │ └─BatchNorm2d: 5-356 [1, 256, 64, 64] [1, 256, 64, 64] |
||
| 425 | │ │ │ │ └─ReLU: 5-357 [1, 256, 64, 64] [1, 256, 64, 64] |
||
| 426 | │ │ │ └─Conv2dReLU: 4-46 [1, 256, 64, 64] [1, 256, 64, 64] |
||
| 427 | │ │ │ │ └─Conv2d: 5-358 [1, 256, 64, 64] [1, 256, 64, 64] |
||
| 428 | │ │ │ │ └─BatchNorm2d: 5-359 [1, 256, 64, 64] [1, 256, 64, 64] |
||
| 429 | │ │ │ │ └─ReLU: 5-360 [1, 256, 64, 64] [1, 256, 64, 64] |
||
| 430 | │ │ │ └─Attention: 4-47 [1, 256, 64, 64] [1, 256, 64, 64] |
||
| 431 | │ │ │ │ └─Identity: 5-361 [1, 256, 64, 64] [1, 256, 64, 64] |
||
| 432 | │ │ └─DecoderBlock: 3-10 [1, 256, 64, 64] [1, 128, 128, 128] |
||
| 433 | │ │ │ └─Attention: 4-48 [1, 296, 128, 128] [1, 296, 128, 128] |
||
| 434 | │ │ │ │ └─Identity: 5-362 [1, 296, 128, 128] [1, 296, 128, 128] |
||
| 435 | │ │ │ └─Conv2dReLU: 4-49 [1, 296, 128, 128] [1, 128, 128, 128] |
||
| 436 | │ │ │ │ └─Conv2d: 5-363 [1, 296, 128, 128] [1, 128, 128, 128] |
||
| 437 | │ │ │ │ └─BatchNorm2d: 5-364 [1, 128, 128, 128] [1, 128, 128, 128] |
||
| 438 | │ │ │ │ └─ReLU: 5-365 [1, 128, 128, 128] [1, 128, 128, 128] |
||
| 439 | │ │ │ └─Conv2dReLU: 4-50 [1, 128, 128, 128] [1, 128, 128, 128] |
||
| 440 | │ │ │ │ └─Conv2d: 5-366 [1, 128, 128, 128] [1, 128, 128, 128] |
||
| 441 | │ │ │ │ └─BatchNorm2d: 5-367 [1, 128, 128, 128] [1, 128, 128, 128] |
||
| 442 | │ │ │ │ └─ReLU: 5-368 [1, 128, 128, 128] [1, 128, 128, 128] |
||
| 443 | │ │ │ └─Attention: 4-51 [1, 128, 128, 128] [1, 128, 128, 128] |
||
| 444 | │ │ │ │ └─Identity: 5-369 [1, 128, 128, 128] [1, 128, 128, 128] |
||
| 445 | │ │ └─DecoderBlock: 3-11 [1, 128, 128, 128] [1, 64, 256, 256] |
||
| 446 | │ │ │ └─Attention: 4-52 [1, 176, 256, 256] [1, 176, 256, 256] |
||
| 447 | │ │ │ │ └─Identity: 5-370 [1, 176, 256, 256] [1, 176, 256, 256] |
||
| 448 | │ │ │ └─Conv2dReLU: 4-53 [1, 176, 256, 256] [1, 64, 256, 256] |
||
| 449 | │ │ │ │ └─Conv2d: 5-371 [1, 176, 256, 256] [1, 64, 256, 256] |
||
| 450 | │ │ │ │ └─BatchNorm2d: 5-372 [1, 64, 256, 256] [1, 64, 256, 256] |
||
| 451 | │ │ │ │ └─ReLU: 5-373 [1, 64, 256, 256] [1, 64, 256, 256] |
||
| 452 | │ │ │ └─Conv2dReLU: 4-54 [1, 64, 256, 256] [1, 64, 256, 256] |
||
| 453 | │ │ │ │ └─Conv2d: 5-374 [1, 64, 256, 256] [1, 64, 256, 256] |
||
| 454 | │ │ │ │ └─BatchNorm2d: 5-375 [1, 64, 256, 256] [1, 64, 256, 256] |
||
| 455 | │ │ │ │ └─ReLU: 5-376 [1, 64, 256, 256] [1, 64, 256, 256] |
||
| 456 | │ │ │ └─Attention: 4-55 [1, 64, 256, 256] [1, 64, 256, 256] |
||
| 457 | │ │ │ │ └─Identity: 5-377 [1, 64, 256, 256] [1, 64, 256, 256] |
||
| 458 | │ │ └─DecoderBlock: 3-12 [1, 64, 256, 256] [1, 32, 512, 512] |
||
| 459 | │ │ │ └─Conv2dReLU: 4-56 [1, 64, 512, 512] [1, 32, 512, 512] |
||
| 460 | │ │ │ │ └─Conv2d: 5-378 [1, 64, 512, 512] [1, 32, 512, 512] |
||
| 461 | │ │ │ │ └─BatchNorm2d: 5-379 [1, 32, 512, 512] [1, 32, 512, 512] |
||
| 462 | │ │ │ │ └─ReLU: 5-380 [1, 32, 512, 512] [1, 32, 512, 512] |
||
| 463 | │ │ │ └─Conv2dReLU: 4-57 [1, 32, 512, 512] [1, 32, 512, 512] |
||
| 464 | │ │ │ │ └─Conv2d: 5-381 [1, 32, 512, 512] [1, 32, 512, 512] |
||
| 465 | │ │ │ │ └─BatchNorm2d: 5-382 [1, 32, 512, 512] [1, 32, 512, 512] |
||
| 466 | │ │ │ │ └─ReLU: 5-383 [1, 32, 512, 512] [1, 32, 512, 512] |
||
| 467 | │ │ │ └─Attention: 4-58 [1, 32, 512, 512] [1, 32, 512, 512] |
||
| 468 | │ │ │ │ └─Identity: 5-384 [1, 32, 512, 512] [1, 32, 512, 512] |
||
| 469 | ├─SegmentationHead: 1-3 [1, 32, 512, 512] [1, 1, 512, 512] |
||
| 470 | │ └─Conv2d: 2-6 [1, 32, 512, 512] [1, 1, 512, 512] |
||
| 471 | │ └─Identity: 2-7 [1, 1, 512, 512] [1, 1, 512, 512] |
||
| 472 | │ └─Activation: 2-8 [1, 1, 512, 512] [1, 1, 512, 512] |
||
| 473 | │ │ └─Identity: 3-13 [1, 1, 512, 512] [1, 1, 512, 512] |
||
| 474 | ==================================================================================================== |