From 328ad4da0402c69563f7112995eec00496a3ea24 Mon Sep 17 00:00:00 2001
From: Glenn Jocher <glenn.jocher@ultralytics.com>
Date: Wed, 19 Feb 2020 18:37:17 -0800
Subject: [PATCH] updates

---
 models.py | 3 +--
 1 file changed, 1 insertion(+), 2 deletions(-)

diff --git a/models.py b/models.py
index 0e9aa117..7a870e0a 100755
--- 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]