🧗

johndpope · Aug 11, 2024 · 45de120 · 45de120
1 parent e94e53c
commit 45de120
Show file tree

Hide file tree

Showing 2 changed files with 1 addition and 1 deletion.
diff --git a/__pycache__/resblock.cpython-311.pyc b/__pycache__/resblock.cpython-311.pyc
diff --git a/model.py b/model.py
@@ -167,7 +167,7 @@ def __init__(self, initial_channels=64, output_channels=[128, 256, 512, 512,512,
         self.res_blocks = nn.ModuleList()
         in_channels = initial_channels
         for out_channels in output_channels:
-            self.res_blocks.append(ResBlock(in_channels, out_channels, downsample=True))
+            self.res_blocks.append(ResBlock(in_channels, out_channels))
             in_channels = out_channels
 
         self.equalconv = EqualConv2d(output_channels[-1], output_channels[-1], kernel_size=3, stride=1, padding=1)