| ========================================================================================== |
| Layer (type:depth-idx) Output Shape Param # |
| ========================================================================================== |
| ImageRegressionNet [1, 3, 640, 1024] -- |
| ├─Sequential: 1-1 [1, 3, 640, 1024] -- |
| │ └─Encoder: 2-1 [1, 3, 640, 1024] -- |
| │ │ └─ConvBlock: 3-1 [1, 9, 640, 1024] 990 |
| │ │ └─ConvBlock: 3-2 [1, 27, 640, 1024] 8,802 |
| │ │ └─Conv2d: 3-3 [1, 3, 640, 1024] 732 |
| │ └─Decoder: 2-2 [1, 3, 640, 1024] -- |
| │ │ └─Upsample: 3-4 [1, 3, 640, 1024] -- |
| ========================================================================================== |
| Total params: 10,524 |
| Trainable params: 10,524 |
| Non-trainable params: 0 |
| Total mult-adds (Units.GIGABYTES): 6.90 |
| ========================================================================================== |
| Input size (MB): 7.86 |
| Forward/backward pass size (MB): 393.22 |
| Params size (MB): 0.04 |
| Estimated Total Size (MB): 401.12 |
| ========================================================================================== |
