fix: extract freture cannot run on pure cpu

2023-04-01 17:01:51 +08:00 · 2023-04-01 17:01:51 +08:00 · 9e59375311
parent 3489b4f6da
commit 9e59375311
1 changed files with 10 additions and 5 deletions
--- a/extract_feature_print.py
+++ b/extract_feature_print.py
@ -1,4 +1,9 @@
 import os,sys,traceback
+if len(sys.argv) == 4:
+    n_part=int(sys.argv[1])
+    i_part=int(sys.argv[2])
+    exp_dir=sys.argv[3]
+else:
    n_part=int(sys.argv[1])
    i_part=int(sys.argv[2])
    i_gpu=sys.argv[3]