Empty val batch CUDA device fix (ultralytics#7539)

glenn-jocher · Clay Januhowski · commit f2cb28e4e974 · 2022-09-08T13:54:09.000-05:00
Verified fix for ultralytics#7525 (comment)
diff --git a/val.py b/val.py
@@ -227,7 +227,7 @@ def run(
 
             if npr == 0:
                 if nl:
-                    stats.append((correct, *torch.zeros((3, 0))))
+                    stats.append((correct, *torch.zeros((3, 0), device=device)))
                 continue
 
             # Predictions
