From cd76a1a9827ec9a2506f272aeae76c13b895746f Mon Sep 17 00:00:00 2001
From: Glenn Jocher <glenn.jocher@ultralytics.com>
Date: Mon, 9 Mar 2020 10:46:59 -0700
Subject: [PATCH] updates

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

diff --git a/models.py b/models.py
index 0b203c82..4c94a676 100755
--- a/models.py
+++ b/models.py
@@ -189,7 +189,7 @@ class YOLOLayer(nn.Module):
         self.ny = 0  # initialize number of y gridpoints
         self.arc = arc
 
-        if ONNX_EXPORT:  # grids must be computed in __init__
+        if ONNX_EXPORT:
             stride = [32, 16, 8][yolo_index]  # stride of this layer
             nx = img_size[1] // stride  # number x grid points
             ny = img_size[0] // stride  # number y grid points