From 894218390b44fa48f19875b4f5b89f23a788158e Mon Sep 17 00:00:00 2001
From: Glenn Jocher <glenn.jocher@ultralytics.com>
Date: Sun, 29 Dec 2019 14:28:56 -0800
Subject: [PATCH] updates

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

diff --git a/models.py b/models.py
index c3bf5a18..c4e751a5 100755
--- a/models.py
+++ b/models.py
@@ -260,8 +260,7 @@ class Darknet(nn.Module):
             elif mtype == 'shortcut':
                 x = x + layer_outputs[int(mdef['from'])]
             elif mtype == 'yolo':
-                x = module(x, img_size)
-                output.append(x)
+                output.append(module(x, img_size))
             layer_outputs.append(x if i in self.routs else [])
 
         if self.training: