ソースを参照

feat: 添加默认配置,简化命令行参数设置

zhch158_admin 3 ヶ月 前
コミット
bc5ee0917b
1 ファイル変更8 行追加0 行削除
  1. 8 0
      zhch/ppstructurev3_scheduler.py

+ 8 - 0
zhch/ppstructurev3_scheduler.py

@@ -562,6 +562,14 @@ if __name__ == "__main__":
             "batch_size": 2,
         }
         
+        default_config = {
+            "input_csv": "./OmniDocBench_Results_Scheduler/processed_files_4procs_20250814_182650.csv",
+            "output_dir": "./OmniDocBench_Results_Scheduler",
+            "num_processes": 4,
+            "devices": "gpu:0,gpu:1,gpu:2,gpu:3",
+            "batch_size": 2,
+        } 
+
         sys.argv = [sys.argv[0]]
         for key, value in default_config.items():
             sys.argv.extend([f"--{key}", str(value)])