Add CPU support

arbellea · arbellea · commit aab3a4d05d6d · 2018-01-01T15:40:16.000+02:00
diff --git a/SourceCode/RNNSeg/Params.py b/SourceCode/RNNSeg/Params.py
@@ -520,6 +520,7 @@ class ParamsEvalBiGRU(ParamsBase):
     save_out_dir = ROOT_SAVE_DIR
 
     # Hardware
+    useGPU = True
     gpu_id = 0
 
     # Net Architecture
