Merge e40201f15f into e4b7543ffa

core/leras/device.py

@@ -68,6 +68,8 @@ class Devices(object):
         return None
 
     def get_devices_from_index_list(self, idx_list):
+        if not isinstance(idx_list, list):
+            idx_list = [int(idx.strip()) for idx in idx_list.split(',')]
         result = []
         for device in self.devices:
             if device.index in idx_list:

main.py

@@ -1,4 +1,5 @@
 if __name__ == "__main__":
+
     # Fix for linux
     import multiprocessing
     multiprocessing.set_start_method("spawn")
@@ -149,7 +150,6 @@ if __name__ == "__main__":
     p.add_argument('--cpu-only', action="store_true", dest="cpu_only", default=False, help="Train on CPU.")
     p.add_argument('--force-gpu-idxs', dest="force_gpu_idxs", default=None, help="Force to choose GPU indexes separated by comma.")
     p.add_argument('--silent-start', action="store_true", dest="silent_start", default=False, help="Silent start. Automatically chooses Best GPU and last used model.")
-    
     p.add_argument('--execute-program', dest="execute_program", default=[], action='append', nargs='+')
     p.set_defaults(func=process_train)
     

requirements-cuda.txt

@@ -1,12 +1,13 @@
-tqdm
-numpy==1.19.3
-numexpr
-h5py==2.10.0
-opencv-python==4.1.0.25
-ffmpeg-python==0.1.17
-scikit-image==0.14.2
-scipy==1.4.1
 colorama
-tensorflow-gpu==2.4.0
+cython==0.29.25
+ffmpeg-python==0.1.17
+h5py==2.10.0
+numpy==1.20.1
+numexpr
+opencv-python==4.1.2.30
 pyqt5
+scikit-image==0.18.3
+scipy==1.10.1
+tensorflow-gpu==2.4.1
 tf2onnx==1.9.3
+tqdm