updates

2020-02-19 18:37:17 -08:00 · 2020-02-19 18:37:17 -08:00 · 328ad4da04
parent 1043832493
commit 328ad4da04
1 changed files with 1 additions and 2 deletions
--- a/models.py
+++ b/models.py
@ -140,8 +140,7 @@ class weightedFeatureFusion(nn.Module):  # weighted sum of 2 or more layers http

            # Adjust channels
            if dc > 0:  # pad
-                pad = nn.ZeroPad2d((0, 0, 0, 0, 0, dc))
-                a = pad(a)
+                a = nn.ZeroPad2d((0, 0, 0, 0, 0, dc))(a)
            elif dc < 0:  # slice
                a = a[:, :nc]