This commit is contained in:
henryruhs
2025-02-12 15:46:33 +01:00
parent 32dfdcf1b3
commit 71e0ae34c0
+1 -1
View File
@@ -44,7 +44,7 @@ class MultiscaleDiscriminator(nn.Module):
class NLayerDiscriminator(nn.Module):
def __init__(self, input_channels : int, num_filters : int, num_layers : int, kernel_size : int) -> None:
super(NLayerDiscriminator, self).__init__()
layers = self.create_layers(input_channels, num_filters, num_layers, kernel_size)
layers = self.create_layers(self, input_channels, num_filters, num_layers, kernel_size)
self.discriminator = nn.Sequential(*layers)
@staticmethod