From bc989a0147746af7ce6af77da949e106bfe94db3 Mon Sep 17 00:00:00 2001
From: Glenn Jocher <glenn.jocher@ultralytics.com>
Date: Tue, 19 Mar 2019 15:44:36 +0200
Subject: [PATCH] multi_gpu multi_scale

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

diff --git a/models.py b/models.py
index 18dc2b50..21e871e1 100755
--- a/models.py
+++ b/models.py
@@ -105,9 +105,8 @@ class YOLOLayer(nn.Module):
         self.nA = len(anchors)  # number of anchors (3)
         self.nC = nC  # number of classes (80)
         self.img_size = 0
-        # self.nG, self.stride, self.grid_xy, self.anchor_vec, self.anchor_wh = \
-        #    [], [], [], [], []
-        create_grids(self, 32, 1)
+        self.nG, self.stride, self.grid_xy, self.anchor_vec, self.anchor_wh = \
+            [], [], [], [], []
 
         if ONNX_EXPORT:  # grids must be computed in __init__
             stride = [32, 16, 8][yolo_layer]  # stride of this layer