|
|
@@ -352,7 +352,7 @@ if __name__ == "__main__":
|
|
|
# "output_dir": "./OmniDocBench_PPStructureV3_Results",
|
|
|
# "collect_results": f"./OmniDocBench_PPStructureV3_Results/processed_files_{time.strftime('%Y%m%d_%H%M%S')}.csv",
|
|
|
"pipeline": "./my_config/PP-StructureV3.yaml",
|
|
|
- "device": "gpu:3",
|
|
|
+ "device": "gpu",
|
|
|
}
|
|
|
|
|
|
# 构造参数
|