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
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
)