From e185719bd795e33ffde909b2de8f36743acac374 Mon Sep 17 00:00:00 2001
From: Glenn Jocher <glenn.jocher@ultralytics.com>
Date: Tue, 4 Feb 2020 21:22:20 -0800
Subject: [PATCH] updates

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

diff --git a/models.py b/models.py
index ea262375..74bfe8e7 100755
--- a/models.py
+++ b/models.py
@@ -268,7 +268,7 @@ class Darknet(nn.Module):
             x = [torch.cat(x, 0) for x in zip(*output)]
             return x[0], torch.cat(x[1:3], 1)  # scores, boxes: 3780x80, 3780x4
         else:
-            io, p = list(zip(*output))  # inference output, training output
+            io, p = zip(*output)  # inference output, training output
             return torch.cat(io, 1), p
 
     def fuse(self):