Disabled extra skip connection

manole-alexandru · Apr 21, 2023 · 3ad56ff · 3ad56ff
1 parent 84841ad
commit 3ad56ff
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/models/common.py b/models/common.py
@@ -854,7 +854,7 @@ def __init__(self, in_channels):
         self.cv1 = Conv(in_channels, 96, k=3)
         self.upsample = nn.Upsample(scale_factor=2, mode='nearest')
         self.cv2 = Conv(192, 48, k=3)
-        self.cv3 = Conv(96, 16, k=3)
+        self.cv3 = Conv(48, 16, k=3)
         self.cv4 = Conv(16, 1, act=False)
         self.relu = nn.ReLU()
 
@@ -867,7 +867,7 @@ def forward(self, x, skipped_input):
 
         x = self.cv2(x)
         x = self.upsample(x)
-        x = torch.cat((x, skipped_input[1]), 1)  # Skip connection
+        # x = torch.cat((x, skipped_input[1]), 1)  # Skip connection
 
         x = self.cv3(x)
         x = self.upsample(x)