| Layer (type) | Output shape | Parameters |
| Conv2d-1 | [−1, 64, 128, 128] | 1,792 | LeakyReLU-2 | [−1, 64, 128, 128] | 0 | Conv2d-3 | [−1, 128, 64, 64] | 73,856 | InstanceNorm2d-4 | [−1, 128, 64, 64] | 0 | LeakyReLU-5 | [−1, 128, 64, 64] | 0 | Conv2d-6 | [−1, 256, 32, 32] | 295,168 | InstanceNorm2d-7 | [−1, 256, 32, 32] | 0 | LeakyReLU-8 | [−1, 256, 32, 32] | 0 | Conv2d-9 | [−1, 512, 31, 31] | 1,180,160 | InstanceNorm2d-10 | [−1, 512, 31, 31] | 0 | LeakyReLU-11 | [−1, 512, 31, 31] | 0 | Conv2d-12 | [−1, 1, 30, 30] | 4,609 |
|
|