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