Ver Fonte

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

zhch158_admin há 3 meses atrás
pai
commit
bc5ee0917b
1 ficheiros alterados com 8 adições e 0 exclusões
  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)])