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

Merge new-arch, because it has proven to give the best results #1

Merged
NikkeDoy merged 14 commits from new-arch into main 2025-04-30 23:47:41 +03:00
Conversation 0 Commits 14 Files Changed 8 +2 -8
2 changed files with 2 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files
Showing only changes of commit 0790a0d3da - Show all commits

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

5
generator.py
View File

@ -18,12 +18,9 @@ class SISUGenerator(nn.Module):
)
self.conv_blocks = nn.Sequential(
conv_block(layer, layer, kernel_size=3, dilation=1), # Local details
conv_block(layer, layer*2, kernel_size=5, dilation=2), # Local Context
conv_block(layer*2, layer*2, kernel_size=3, dilation=4), # Wider context
conv_block(layer*2, layer*4, kernel_size=7, dilation=8), # Longer range dependencies
conv_block(layer, layer*4, kernel_size=5, dilation=2), # Local Context
conv_block(layer*4, layer*4, kernel_size=3, dilation=16), # Longer range dependencies
conv_block(layer*4, layer*2, kernel_size=5, dilation=8), # Wider context
conv_block(layer*2, layer*2, kernel_size=3, dilation=4), # Wider context
conv_block(layer*2, layer, kernel_size=5, dilation=2), # Local Context
conv_block(layer, layer, kernel_size=3, dilation=1), # Local details
)