| 61 | |
| 62 | |
| 63 | class Upsample2(nn.Module): |
| 64 | def __init__(self, scale_factor, mode='nearest'): |
| 65 | super().__init__() |
| 66 | self.factor = scale_factor |
| 67 | self.mode = mode |
| 68 | |
| 69 | def forward(self, x): |
| 70 | return torch.nn.functional.interpolate(x, scale_factor=self.factor, mode=self.mode) |
| 71 | |
| 72 | |
| 73 | class Upsample(nn.Module): |