==========================================================================================
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
==========================================================================================