Spaces:

xiang-wuu
/

yolov5

Runtime error

glenn-jocher commited on Apr 22, 2022

Commit

813eba8

unverified ·

1 Parent(s): b804b36

Empty val batch CUDA device fix (#7539)

Verified fix for https://github.com/ultralytics/yolov5/pull/7525#issuecomment-1106081123

Files changed (1) hide show

val.py CHANGED Viewed

@@ -227,7 +227,7 @@ def run(
             if npr == 0:
                 if nl:
-                    stats.append((correct, *torch.zeros((3, 0))))
                 continue
             # Predictions

             if npr == 0:
                 if nl:
+                    stats.append((correct, *torch.zeros((3, 0), device=device)))
                 continue
             # Predictions