From 3cf8a13910b26eacbdc5e9bdf3965648b807d149 Mon Sep 17 00:00:00 2001 From: Glenn Jocher Date: Sat, 22 Feb 2020 12:56:20 -0800 Subject: [PATCH] updates --- models.py | 7 +------ 1 file changed, 1 insertion(+), 6 deletions(-) diff --git a/models.py b/models.py index 6a3a4ea7..9214bfc2 100755 --- a/models.py +++ b/models.py @@ -141,15 +141,10 @@ class weightedFeatureFusion(nn.Module): # weighted sum of 2 or more layers http # Adjust channels if dc > 0: # slice input - # a = nn.ZeroPad2d((0, 0, 0, 0, 0, dc))(a) - x[:, :ac] = x[:, :ac] + a + x[:, :ac] = x[:, :ac] + a # or a = nn.ZeroPad2d((0, 0, 0, 0, 0, dc))(a); x = x + a elif dc < 0: # slice feature - if self.n == 2: - return x + a[:, :nc] x = x + a[:, :nc] else: # same shape - if self.n == 2: - return x + a x = x + a return x