NikkeDoy/SISU
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

⚗️ | Experiment with other layer layouts.

Browse Source
This commit is contained in:
NikkeDoy
2024-12-21 23:27:38 +02:00
parent b7d7e95c89
commit 70e20f53d4
6 changed files with 142 additions and 91 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
discriminator.py
View File

@ -5,15 +5,15 @@ class SISUDiscriminator(nn.Module):
super(SISUDiscriminator, self).__init__()
self.model = nn.Sequential(
nn.Conv1d(2, 128, kernel_size=3, padding=1),
nn.LeakyReLU(0.2, inplace=True),
#nn.LeakyReLU(0.2, inplace=True),
nn.Conv1d(128, 256, kernel_size=3, padding=1),
nn.LeakyReLU(0.2, inplace=True),
nn.Conv1d(256, 128, kernel_size=3, padding=1),
nn.LeakyReLU(0.2, inplace=True),
#nn.LeakyReLU(0.2, inplace=True),
nn.Conv1d(128, 64, kernel_size=3, padding=1),
nn.LeakyReLU(0.2, inplace=True),
#nn.LeakyReLU(0.2, inplace=True),
nn.Conv1d(64, 1, kernel_size=3, padding=1),
nn.LeakyReLU(0.2, inplace=True),
#nn.LeakyReLU(0.2, inplace=True),
)
self.global_avg_pool = nn.AdaptiveAvgPool1d(1) # Output size (1,)