updates

2020-03-09 10:46:59 -07:00 · 2020-03-09 10:46:59 -07:00 · cd76a1a982
parent 071d4113f6
commit cd76a1a982
1 changed files with 1 additions and 1 deletions
--- 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