From cf462429d447f8c9fe6ffd1ac3ea41e5622073e4 Mon Sep 17 00:00:00 2001
From: Glenn Jocher <glenn.jocher@ultralytics.com>
Date: Thu, 26 Sep 2019 12:54:09 +0200
Subject: [PATCH] updates

---
 utils/torch_utils.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/utils/torch_utils.py b/utils/torch_utils.py
index ef87d6ca..7c099875 100644
--- a/utils/torch_utils.py
+++ b/utils/torch_utils.py
@@ -15,6 +15,7 @@ def init_seeds(seed=0):
 
 
 def select_device(device='', apex=False):
+    # device = '' or 'cpu' or '0' or '0,1'
     if device.lower() == 'cpu':
         pass
     elif device:  # Set environment variable if device is specified