(self, input_nc, output_nc, norm_layer=nn.BatchNorm2d, nonlinearity=nn.LeakyReLU(), use_spect=False)
| 293 | """ |
| 294 | |
| 295 | def __init__(self, input_nc, output_nc, norm_layer=nn.BatchNorm2d, nonlinearity=nn.LeakyReLU(), use_spect=False): |
| 296 | super(FirstBlock2d, self).__init__() |
| 297 | kwargs = {'kernel_size': 7, 'stride': 1, 'padding': 3} |
| 298 | conv = spectral_norm(nn.Conv2d(input_nc, output_nc, **kwargs), use_spect) |
| 299 | |
| 300 | if type(norm_layer) == type(None): |
| 301 | self.model = nn.Sequential(conv, nonlinearity) |
| 302 | else: |
| 303 | self.model = nn.Sequential(conv, norm_layer(output_nc), nonlinearity) |
| 304 | |
| 305 | def forward(self, x): |
| 306 | out = self.model(x) |
nothing calls this directly
no test coverage detected