mirror of
https://github.com/facefusion/facefusion-labs.git
synced 2026-07-28 16:08:50 +02:00
Follow the concept of layers and sequences
This commit is contained in:
@@ -25,7 +25,7 @@ class Discriminator(nn.Module):
|
||||
discriminators = nn.ModuleList()
|
||||
|
||||
for _ in range(num_discriminators):
|
||||
discriminator = NLD(input_channels, num_filters, num_layers, kernel_size).nld
|
||||
discriminator = NLD(input_channels, num_filters, num_layers, kernel_size).sequences
|
||||
discriminators.append(discriminator)
|
||||
|
||||
return discriminators
|
||||
|
||||
Reference in New Issue
Block a user