@@ -334,7 +334,7 @@ if __name__ == "__main__":
"output_dir": "./OmniDocBench_Results_Single",
"pipeline": "PP-StructureV3",
"device": "gpu:0",
- "batch_size": 4,
+ "batch_size": 2,
}
# 构造参数