(self, num_block, input_nc, feature_nc, norm_layer=nn.BatchNorm2d, nonlinearity=nn.LeakyReLU(),
use_spect=False)
| 342 | |
| 343 | class FineADAINResBlocks(nn.Module): |
| 344 | def __init__(self, num_block, input_nc, feature_nc, norm_layer=nn.BatchNorm2d, nonlinearity=nn.LeakyReLU(), |
| 345 | use_spect=False): |
| 346 | super(FineADAINResBlocks, self).__init__() |
| 347 | self.num_block = num_block |
| 348 | for i in range(num_block): |
| 349 | model = FineADAINResBlock2d(input_nc, feature_nc, norm_layer, nonlinearity, use_spect) |
| 350 | setattr(self, 'res' + str(i), model) |
| 351 | |
| 352 | def forward(self, x, z): |
| 353 | for i in range(self.num_block): |
nothing calls this directly
no test coverage detected