From f2cb28e4e974b85936788dcfa54b32bb8a4c2a5c Mon Sep 17 00:00:00 2001
From: Glenn Jocher <glenn.jocher@ultralytics.com>
Date: Fri, 22 Apr 2022 12:01:14 -0700
Subject: [PATCH] Empty val batch CUDA device fix (#7539)

Verified fix for https://github.com/ultralytics/yolov5/pull/7525#issuecomment-1106081123
---
 val.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/val.py b/val.py
index b2b3bc75911e..58113f016a58 100644
--- 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