:albemic: | Tests.

2025-03-25 19:50:51 +02:00
parent 7e1c7e935a
commit 54338e55a9
4 changed files with 21 additions and 17 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=4, num_rirb=4): #increased base layer and rirb amounts
+    def __init__(self, layer=64, 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),