From 2f1c9a3f6ffb0339334046b4b096b19f78850c0c Mon Sep 17 00:00:00 2001 From: Glenn Jocher Date: Sun, 24 Nov 2019 18:31:06 -1000 Subject: [PATCH] updates --- utils/torch_utils.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/utils/torch_utils.py b/utils/torch_utils.py index 466a22ae..e7e15715 100644 --- a/utils/torch_utils.py +++ b/utils/torch_utils.py @@ -33,7 +33,7 @@ def select_device(device='', apex=False, batch_size=None): if i == 1: s = ' ' * len(s) print("%sdevice%g _CudaDeviceProperties(name='%s', total_memory=%dMB)" % - (cuda_str, i, x[i].name, x[i].total_memory / c)) + (s, i, x[i].name, x[i].total_memory / c)) else: print('Using CPU')