| | Layer (type) | Output shape | Parameters |
| | ReflectionPad2d-1 | [−1, 3, 262, 262] | 0 | | Conv2d-2 | [−1, 64, 256, 256] | 1,792 | | InstanceNorm2d-3 | [−1, 64, 256, 256] | 0 | | ReLU-4 | [−1, 64, 256, 256] | 0 | | Conv2d-5 | [−1, 128, 128, 128] | 73,856 | | InstanceNorm2d-6 | [−1, 128, 128, 128] | 0 | | ReLU-7 | [−1, 128, 128, 128] | 0 | | Conv2d-8 | [−1, 256, 64, 64] | 295,168 | | InstanceNorm2d-9 | [−1, 256, 64, 64] | 0 | | ReLU-10 | [−1, 256, 64, 64] | 0 | | ReflectionPad2d-11 | [−1, 256, 66, 66] | 0 | | Conv2d-12 | [−1, 256, 64, 64] | 590,080 | | ReLU-13 | [−1, 256, 64, 64] | 0 | | ReflectionPad2d-14 | [−1, 256, 66, 66] | 0 | | Conv2d-15 | [−1, 256, 64, 64] | 590,080 | | ResidualBlock-16 | [−1, 256, 64, 64] | 0 | | ConvTranspose2d-65 | [−1, 128, 128, 128] | 295,040 | | InstanceNorm2d-66 | [−1, 128, 128, 128] | 0 | | ReLU-67 | [−1, 128, 128, 128] | 0 | | ConvTranspose2d-68 | [−1, 64, 256, 256] | 73,792 | | InstanceNorm2d-69 | [−1, 64, 256, 256] | 0 | | ReLU-70 | [−1, 64, 256, 256] | 0 | | ReflectionPad2d-71 | [−1, 64, 262, 262] | 0 | | Conv2d-72 | [−1, 3, 256, 256] | 1,731 | | Tanh-73 | [−1, 3, 256, 256] | 0 |
|
|