From d5b5f74167aeae644e3558ec4da172246e4e0d88 Mon Sep 17 00:00:00 2001
From: Glenn Jocher <glenn.jocher@ultralytics.com>
Date: Fri, 13 Sep 2019 16:27:15 +0200
Subject: [PATCH] updates

---
 utils/torch_utils.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/utils/torch_utils.py b/utils/torch_utils.py
index 6b4b2624..11b3504e 100644
--- a/utils/torch_utils.py
+++ b/utils/torch_utils.py
@@ -17,12 +17,12 @@ def init_seeds(seed=0):
 
 def select_device(device=None, apex=False):
     if device == 'cpu':
-        force_cpu = True
+        pass
     elif device:  # Set environment variable if device is specified
         os.environ['CUDA_VISIBLE_DEVICES'] = device
 
     # apex if mixed precision training https://github.com/NVIDIA/apex
-    cuda = False if force_cpu else torch.cuda.is_available()
+    cuda = False if device == 'cpu' else torch.cuda.is_available()
     device = torch.device('cuda:0' if cuda else 'cpu')
 
     if not cuda: