: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/generator.py
+++ b/generator.py
@@ -34,7 +34,7 @@ class ResidualInResidualBlock(nn.Module):
        return x + residual

 class SISUGenerator(nn.Module):
-    def __init__(self, layer=64, num_rirb=4): #increased base layer and rirb amounts
+    def __init__(self, layer=4, num_rirb=4): #increased base layer and rirb amounts
        super(SISUGenerator, self).__init__()
        self.conv1 = nn.Sequential(
            nn.Conv1d(1, layer, kernel_size=7, padding=3),