From 67e7ac221fcb36d75579bc9269d72b7b06286506 Mon Sep 17 00:00:00 2001
From: Glenn Jocher <glenn.jocher@ultralytics.com>
Date: Mon, 9 Mar 2020 14:20:38 -0700
Subject: [PATCH] updates

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

diff --git a/models.py b/models.py
index 4c94a676..80976a66 100755
--- a/models.py
+++ b/models.py
@@ -195,7 +195,7 @@ class YOLOLayer(nn.Module):
             ny = img_size[0] // stride  # number y grid points
             create_grids(self, img_size, (nx, ny))
 
-    def forward(self, p, img_size, var=None):
+    def forward(self, p, img_size):
         if ONNX_EXPORT:
             bs = 1  # batch size
         else:
@@ -261,10 +261,9 @@ class Darknet(nn.Module):
         self.version = np.array([0, 2, 5], dtype=np.int32)  # (int32) version info: major, minor, revision
         self.seen = np.array([0], dtype=np.int64)  # (int64) number of images seen during training
 
-    def forward(self, x, var=None):
+    def forward(self, x, verbose=False):
         img_size = x.shape[-2:]
         yolo_out, out = [], []
-        verbose = False
         if verbose:
             str = ''
             print('0', x.shape)