:albemic: | More tests.

2025-03-25 21:51:29 +02:00
parent 54338e55a9
commit f928d8c2cf
3 changed files with 14 additions and 12 deletions
--- a/discriminator.py
+++ b/discriminator.py
@ -28,7 +28,7 @@ class AttentionBlock(nn.Module):
        return x * attention_weights

 class SISUDiscriminator(nn.Module):
-    def __init__(self, layers=64): #Increased base layer count
+    def __init__(self, layers=4): #Increased base layer count
        super(SISUDiscriminator, self).__init__()
        self.model = nn.Sequential(
            discriminator_block(1, layers, kernel_size=3, stride=1), #Aggressive downsampling