This commit is contained in:
Glenn Jocher
2019-07-23 15:08:28 +02:00
parent 308eda38fd
commit b025b3123e
2 changed files with 2 additions and 2 deletions
+1 -1
View File
@@ -18,7 +18,7 @@ def select_device(force_cpu=False):
if cuda:
try: # Mixed precision training https://github.com/NVIDIA/apex
from apex import amp
apex_str = 'with Apex '
apex_str = 'Apex '
except:
apex_str = ''