From dd320ea5bed3f67a1df31241f40446e8dc4a64d1 Mon Sep 17 00:00:00 2001 From: henryruhs Date: Wed, 12 Feb 2025 15:47:04 +0100 Subject: [PATCH] Fix self --- face_swapper/src/models/discriminator.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/face_swapper/src/models/discriminator.py b/face_swapper/src/models/discriminator.py index 5a2865c..786b2a9 100644 --- a/face_swapper/src/models/discriminator.py +++ b/face_swapper/src/models/discriminator.py @@ -44,11 +44,11 @@ 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(self, input_channels, num_filters, num_layers, kernel_size) + layers = self.create_layers(input_channels, num_filters, num_layers, kernel_size) self.discriminator = nn.Sequential(*layers) @staticmethod - def create_layers(self, input_channels : int, num_filters : int, num_layers: int, kernel_size : int) -> List[nn.Module]: + def create_layers(input_channels : int, num_filters : int, num_layers: int, kernel_size : int) -> List[nn.Module]: padding = int(numpy.ceil((kernel_size - 1) / 2)) current_filters = num_filters layers =\