| 483 | |
| 484 | |
| 485 | class Upsample2D(nn.Module): |
| 486 | |
| 487 | def __init__(self, channels): |
| 488 | super().__init__() |
| 489 | self.conv = nn.Conv2d(channels, channels, 3, padding=1) |
| 490 | |
| 491 | def forward(self, x): |
| 492 | x = F.interpolate(x, scale_factor=2.0, mode="nearest") |
| 493 | return self.conv(x) |
| 494 | |
| 495 | |
| 496 | class Downsample2D(nn.Module): |