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

⚗️ | Experimenting with smaller architecture.

Browse Source
This commit is contained in:
NikkeDoy
2025-01-25 16:48:10 +02:00
parent f615b39ded
commit 0790a0d3da
2 changed files with 2 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
discriminator.py
View File

@ -20,12 +20,9 @@ class SISUDiscriminator(nn.Module):
# Core Discriminator Blocks with varied kernels and dilations
discriminator_block(layers, layers * 2, kernel_size=5, stride=2, dilation=1), # Downsample
discriminator_block(layers * 2, layers * 2, kernel_size=3, dilation=2),
discriminator_block(layers * 2, layers * 4, kernel_size=5, dilation=4),
discriminator_block(layers * 4, layers * 4, kernel_size=3, dilation=8),
discriminator_block(layers * 4, layers * 8, kernel_size=5, dilation=16),
discriminator_block(layers * 8, layers * 8, kernel_size=3, dilation=8),
discriminator_block(layers * 8, layers * 4, kernel_size=5, dilation=4),
discriminator_block(layers * 8, layers * 4, kernel_size=3, dilation=8),
discriminator_block(layers * 4, layers * 2, kernel_size=3, dilation=2),
discriminator_block(layers * 2, layers, kernel_size=5, dilation=1),
# Final Convolution