|
|
@@ -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)])
|