Bienqq
/
car-detection-bayes
2
1
Fork 0
Code Issues Pull Requests Releases Wiki Activity

updates

This commit is contained in:
Glenn Jocher 2019-11-24 18:31:06 -10:00
parent f12a2a513a
commit 2f1c9a3f6f
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
utils/torch_utils.py
View File

@ -33,7 +33,7 @@ def select_device(device='', apex=False, batch_size=None):
if i == 1: if i == 1:
s = ' ' * len(s) s = ' ' * len(s)
print("%sdevice%g _CudaDeviceProperties(name='%s', total_memory=%dMB)" % 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: else:
print('Using CPU') print('Using CPU')