Fix bug in val_onnx where grid strides were consumed by generator (#93)…

… (#94)
neuralmagic · Aug 22, 2022 · d2863d2 · d2863d2
1 parent ca7bd4d
commit d2863d2
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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))
     )