mirror of
https://github.com/facefusion/facefusion-labs.git
synced 2026-04-19 15:56:37 +02:00
Fix self
This commit is contained in:
@@ -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
|
||||
|
||||
Reference in New Issue
Block a user