From d581707b45cfd6494da175e94871e72442693a2e Mon Sep 17 00:00:00 2001
From: corey-nm <109536191+corey-nm@users.noreply.github.com>
Date: Mon, 22 Aug 2022 10:47:59 -0400
Subject: [PATCH] Fix bug in val_onnx where grid strides were consumed by
 generator (#93)

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

diff --git a/val_onnx.py b/val_onnx.py
index f18bad612792..2df802f6f680 100644
--- a/val_onnx.py
+++ b/val_onnx.py
@@ -155,7 +155,7 @@ def get_stride(yolo_pipeline: Pipeline) -> int:
         if not isinstance(image_size, int):
             image_size = image_size[0]
 
-    grid_shapes = (
+    grid_shapes = list(
         get_tensor_dim_shape(model.graph.output[index], 2)
         for index in range(1, len(model.graph.output))
     )