----------------------------------------------------------------
        Layer (type)               Output Shape         Param #
================================================================
            Conv2d-1         [-1, 16, 224, 224]           5,200
       BatchNorm2d-2         [-1, 16, 224, 224]              32
            Conv2d-3         [-1, 32, 112, 112]           8,224
       BatchNorm2d-4         [-1, 32, 112, 112]              64
            Conv2d-5           [-1, 64, 56, 56]          32,832
       BatchNorm2d-6           [-1, 64, 56, 56]             128
            Conv2d-7           [-1, 64, 56, 56]          36,928
       BatchNorm2d-8           [-1, 64, 56, 56]             128
            Conv2d-9           [-1, 64, 56, 56]          36,928
      BatchNorm2d-10           [-1, 64, 56, 56]             128
    ResidualBlock-11           [-1, 64, 56, 56]               0
           Conv2d-12           [-1, 64, 56, 56]          36,928
      BatchNorm2d-13           [-1, 64, 56, 56]             128
           Conv2d-14           [-1, 64, 56, 56]          36,928
      BatchNorm2d-15           [-1, 64, 56, 56]             128
    ResidualBlock-16           [-1, 64, 56, 56]               0
           Conv2d-17           [-1, 64, 56, 56]          36,928
      BatchNorm2d-18           [-1, 64, 56, 56]             128
           Conv2d-19           [-1, 64, 56, 56]          36,928
      BatchNorm2d-20           [-1, 64, 56, 56]             128
    ResidualBlock-21           [-1, 64, 56, 56]               0
           Conv2d-22           [-1, 64, 56, 56]          36,928
      BatchNorm2d-23           [-1, 64, 56, 56]             128
           Conv2d-24           [-1, 64, 56, 56]          36,928
      BatchNorm2d-25           [-1, 64, 56, 56]             128
    ResidualBlock-26           [-1, 64, 56, 56]               0
           Conv2d-27           [-1, 64, 56, 56]          36,928
      BatchNorm2d-28           [-1, 64, 56, 56]             128
           Conv2d-29           [-1, 64, 56, 56]          36,928
      BatchNorm2d-30           [-1, 64, 56, 56]             128
    ResidualBlock-31           [-1, 64, 56, 56]               0
  ConvTranspose2d-32         [-1, 32, 113, 113]          32,800
      BatchNorm2d-33         [-1, 32, 113, 113]              64
  ConvTranspose2d-34         [-1, 16, 225, 225]           8,208
      BatchNorm2d-35         [-1, 16, 225, 225]              32
  ConvTranspose2d-36         [-1, 16, 225, 225]          20,752
          Dropout-37         [-1, 16, 225, 225]               0
           Conv2d-38          [-1, 1, 224, 224]              65
          Dropout-39         [-1, 16, 225, 225]               0
           Conv2d-40          [-1, 1, 224, 224]              65
          Dropout-41         [-1, 16, 225, 225]               0
           Conv2d-42          [-1, 1, 224, 224]              65
          Dropout-43         [-1, 16, 225, 225]               0
           Conv2d-44          [-1, 1, 224, 224]              65
================================================================
Total params: 479,156
Trainable params: 479,156
Non-trainable params: 0
----------------------------------------------------------------
Input size (MB): 0.77
Forward/backward pass size (MB): 110.74
Params size (MB): 1.83
Estimated Total Size (MB): 113.34
----------------------------------------------------------------