From 3ad56ff9c81c5262bf0fad2799a9978e72769b66 Mon Sep 17 00:00:00 2001 From: ManoleAlexandru99 Date: Fri, 21 Apr 2023 20:55:20 +0300 Subject: [PATCH] Disabled extra skip connection --- models/common.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/models/common.py b/models/common.py index ce8417d3f837..85e047906b50 100644 --- 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)