Merge pull request #97 from amcadmus/devel

Devel

Author: amcadmus

source/train/Local.py

@@ -5,6 +5,9 @@ def get_resource ():
     nodelist = [nodename]
     gpus = os.getenv('CUDA_VISIBLE_DEVICES')
     if gpus is not None :
-        gpus = gpus.split(",")
-        gpus = [int(ii) for ii in gpus]
+        if gpus != "" :
+            gpus = gpus.split(",")
+            gpus = [int(ii) for ii in gpus]
+        else :
+            gpus = []
     return nodename, nodelist, gpus