LearningToPaint: fix cuda bias code

pytorch · Dec 11, 2023 · df975be · df975be
1 parent 7de2aed
commit df975be
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/torchbenchmark/models/LearningToPaint/baseline/utils/util.py b/torchbenchmark/models/LearningToPaint/baseline/utils/util.py
@@ -14,7 +14,7 @@ def prLightGray(prt): print("\033[97m {}\033[00m" .format(prt))
 def prBlack(prt): print("\033[98m {}\033[00m" .format(prt))
 
 def to_numpy(var):
-    return var.cpu().data.numpy() if USE_CUDA else var.data.numpy()
+    return var.cpu().data.numpy()
 
 def to_tensor(ndarray, device):
     return torch.tensor(ndarray, dtype=torch.float, device=device)