From f5418615337a0a6a66b649db68952a3df70e3dc6 Mon Sep 17 00:00:00 2001 From: Glenn Jocher Date: Mon, 25 Feb 2019 13:50:01 +0100 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 6d10004c..2eb65448 100644 --- a/utils/torch_utils.py +++ b/utils/torch_utils.py @@ -20,5 +20,5 @@ def select_device(force_cpu=False): torch.cuda.set_device(0) # OPTIONAL: Set your GPU if multiple available # print('Using ', torch.cuda.device_count(), ' GPUs') - print('Using %s\n%s' % (device.type, torch.cuda.get_device_properties(0) if cuda else '')) + print('Using %s %s\n' % (device.type, torch.cuda.get_device_properties(0) if cuda else '')) return device