From 0790a0d3dab77d071ba5d8c3bc7f68b7845a11c4 Mon Sep 17 00:00:00 2001 From: NikkeDoy Date: Sat, 25 Jan 2025 16:48:10 +0200 Subject: [PATCH] :alembic: | Experimenting with smaller architecture. --- discriminator.py | 5 +---- generator.py | 5 +---- 2 files changed, 2 insertions(+), 8 deletions(-) diff --git a/discriminator.py b/discriminator.py index af29f5d..b800eda 100644 --- a/discriminator.py +++ b/discriminator.py @@ -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 diff --git a/generator.py b/generator.py index 6ea267d..2446275 100644 --- a/generator.py +++ b/generator.py @@ -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 )