Downloads: 1
Diff of
/model-description/decoder.txt
[000000]
..
[637b40]
Switch to unified view
| a | b/model-description/decoder.txt | ||
|---|---|---|---|
| 1 | ========================================================================================== |
||
| 2 | Layer (type:depth-idx) Input Shape Output Shape |
||
| 3 | ========================================================================================== |
||
| 4 | UnetDecoder -- -- |
||
| 5 | ├─ModuleList: 1-1 -- -- |
||
| 6 | ├─Identity: 1-2 [1, 512, 16, 16] [1, 512, 16, 16] |
||
| 7 | ├─ModuleList: 1-1 -- -- |
||
| 8 | │ └─DecoderBlock: 2-1 [1, 512, 16, 16] [1, 512, 32, 32] |
||
| 9 | │ │ └─Attention: 3-1 [1, 688, 32, 32] [1, 688, 32, 32] |
||
| 10 | │ │ └─Conv2dReLU: 3-2 [1, 688, 32, 32] [1, 512, 32, 32] |
||
| 11 | │ │ └─Conv2dReLU: 3-3 [1, 512, 32, 32] [1, 512, 32, 32] |
||
| 12 | │ │ └─Attention: 3-4 [1, 512, 32, 32] [1, 512, 32, 32] |
||
| 13 | │ └─DecoderBlock: 2-2 [1, 512, 32, 32] [1, 256, 64, 64] |
||
| 14 | │ │ └─Attention: 3-5 [1, 576, 64, 64] [1, 576, 64, 64] |
||
| 15 | │ │ └─Conv2dReLU: 3-6 [1, 576, 64, 64] [1, 256, 64, 64] |
||
| 16 | │ │ └─Conv2dReLU: 3-7 [1, 256, 64, 64] [1, 256, 64, 64] |
||
| 17 | │ │ └─Attention: 3-8 [1, 256, 64, 64] [1, 256, 64, 64] |
||
| 18 | │ └─DecoderBlock: 2-3 [1, 256, 64, 64] [1, 128, 128, 128] |
||
| 19 | │ │ └─Attention: 3-9 [1, 296, 128, 128] [1, 296, 128, 128] |
||
| 20 | │ │ └─Conv2dReLU: 3-10 [1, 296, 128, 128] [1, 128, 128, 128] |
||
| 21 | │ │ └─Conv2dReLU: 3-11 [1, 128, 128, 128] [1, 128, 128, 128] |
||
| 22 | │ │ └─Attention: 3-12 [1, 128, 128, 128] [1, 128, 128, 128] |
||
| 23 | │ └─DecoderBlock: 2-4 [1, 128, 128, 128] [1, 64, 256, 256] |
||
| 24 | │ │ └─Attention: 3-13 [1, 176, 256, 256] [1, 176, 256, 256] |
||
| 25 | │ │ └─Conv2dReLU: 3-14 [1, 176, 256, 256] [1, 64, 256, 256] |
||
| 26 | │ │ └─Conv2dReLU: 3-15 [1, 64, 256, 256] [1, 64, 256, 256] |
||
| 27 | │ │ └─Attention: 3-16 [1, 64, 256, 256] [1, 64, 256, 256] |
||
| 28 | │ └─DecoderBlock: 2-5 [1, 64, 256, 256] [1, 32, 512, 512] |
||
| 29 | │ │ └─Conv2dReLU: 3-17 [1, 64, 512, 512] [1, 32, 512, 512] |
||
| 30 | │ │ └─Conv2dReLU: 3-18 [1, 32, 512, 512] [1, 32, 512, 512] |
||
| 31 | │ │ └─Attention: 3-19 [1, 32, 512, 512] [1, 32, 512, 512] |
||
| 32 | ========================================================================================== |