Learned 2x upsampling without padding
| 120 | return x |
| 121 | |
| 122 | class TransposedUpsample(nn.Module): |
| 123 | 'Learned 2x upsampling without padding' |
| 124 | def __init__(self, channels, out_channels=None, ks=5): |
| 125 | super().__init__() |
| 126 | self.channels = channels |
| 127 | self.out_channels = out_channels or channels |
| 128 | |
| 129 | self.up = nn.ConvTranspose2d(self.channels,self.out_channels,kernel_size=ks,stride=2) |
| 130 | |
| 131 | def forward(self,x): |
| 132 | return self.up(x) |
| 133 | |
| 134 | |
| 135 | class Downsample(nn.Module): |
nothing calls this directly
no outgoing calls
no test coverage detected