⚗️ | Experimenting with training optimization.

2024-12-21 00:24:00 +02:00
parent 1fa2a13091
commit b7d7e95c89
4 changed files with 85 additions and 52 deletions
--- a/generator.py
+++ b/generator.py
@@ -5,13 +5,18 @@ class SISUGenerator(nn.Module):
        super(SISUGenerator, self).__init__()
        self.model = nn.Sequential(
            nn.Conv1d(2, 128, kernel_size=3, padding=1),
+            nn.LeakyReLU(0.2, inplace=True),
            nn.Conv1d(128, 256, kernel_size=3, padding=1),
+            nn.LeakyReLU(0.2, inplace=True),

            nn.Upsample(scale_factor=upscale_scale, mode='nearest'),

            nn.Conv1d(256, 128, kernel_size=3, padding=1),
+            nn.LeakyReLU(0.2, inplace=True),
            nn.Conv1d(128, 64, kernel_size=3, padding=1),
-            nn.Conv1d(64, 2, kernel_size=3, padding=1)
+            nn.LeakyReLU(0.2, inplace=True),
+            nn.Conv1d(64, 2, kernel_size=3, padding=1),
+            nn.Tanh()
        )

    def forward(self, x):