Follow the concept of layers and sequences

This commit is contained in:
henryruhs
2025-03-11 14:43:10 +01:00
parent 58a85a80bb
commit cadbe9cf76
2 changed files with 8 additions and 7 deletions
+1 -1
View File
@@ -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