diff --git a/face_swapper/src/models/discriminator.py b/face_swapper/src/models/discriminator.py index 786b2a9..0cc51d4 100644 --- a/face_swapper/src/models/discriminator.py +++ b/face_swapper/src/models/discriminator.py @@ -26,7 +26,7 @@ class MultiscaleDiscriminator(nn.Module): for _ in range(num_discriminators): discriminator = NLayerDiscriminator(input_channels, num_filters, num_layers, kernel_size).discriminator - self.discriminators.append(discriminator) + discriminators.append(discriminator) return discriminators