From df975be7890fa174d21f738c4547e44dd17a6521 Mon Sep 17 00:00:00 2001 From: weishi-deng Date: Sun, 10 Dec 2023 23:03:28 -0800 Subject: [PATCH] LearningToPaint: fix cuda bias code --- torchbenchmark/models/LearningToPaint/baseline/utils/util.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/torchbenchmark/models/LearningToPaint/baseline/utils/util.py b/torchbenchmark/models/LearningToPaint/baseline/utils/util.py index 217e1b9f68..22fe16cf55 100644 --- 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)