Преглед на файлове

Added warning if cuda is disabled, to avoid silent failure

Maxime Chevalier-Boisvert преди 7 години
родител
ревизия
d4d83a9bf6
променени са 1 файла, в които са добавени 3 реда и са изтрити 0 реда
  1. 3 0
      pytorch-rl/arguments.py

+ 3 - 0
pytorch-rl/arguments.py

@@ -64,4 +64,7 @@ def get_args():
     args.cuda = not args.no_cuda and torch.cuda.is_available()
     args.vis = not args.no_vis
 
+    if not args.cuda:
+        print('*** WARNING: CUDA NOT ENABLED ***')
+
     return args