(self, image_nc, feature_nc, ngf, img_f, layers, num_block, norm_layer=nn.BatchNorm2d, nonlinearity=nn.LeakyReLU(), use_spect=False)
| 215 | class FineDecoder(nn.Module): |
| 216 | """docstring for FineDecoder""" |
| 217 | def __init__(self, image_nc, feature_nc, ngf, img_f, layers, num_block, norm_layer=nn.BatchNorm2d, nonlinearity=nn.LeakyReLU(), use_spect=False): |
| 218 | super(FineDecoder, self).__init__() |
| 219 | self.layers = layers |
| 220 | for i in range(layers)[::-1]: |
| 221 | in_channels = min(ngf*(2**(i+1)), img_f) |
| 222 | out_channels = min(ngf*(2**i), img_f) |
| 223 | up = UpBlock2d(in_channels, out_channels, norm_layer, nonlinearity, use_spect) |
| 224 | res = FineADAINResBlocks(num_block, in_channels, feature_nc, norm_layer, nonlinearity, use_spect) |
| 225 | jump = Jump(out_channels, norm_layer, nonlinearity, use_spect) |
| 226 | |
| 227 | setattr(self, 'up' + str(i), up) |
| 228 | setattr(self, 'res' + str(i), res) |
| 229 | setattr(self, 'jump' + str(i), jump) |
| 230 | |
| 231 | self.final = FinalBlock2d(out_channels, image_nc, use_spect, 'tanh') |
| 232 | |
| 233 | self.output_nc = out_channels |
| 234 | |
| 235 | def forward(self, x, z): |
| 236 | out = x.pop() |
nothing calls this directly
no test coverage detected