|
|
@@ -50,7 +50,7 @@ def worker(pipeline_name_or_config_path: str,
|
|
|
os.environ['CUDA_VISIBLE_DEVICES'] = device_id
|
|
|
|
|
|
# 设置paddle使用单精度,避免混合精度问题
|
|
|
- paddle.set_default_dtype("float32")
|
|
|
+ # paddle.set_default_dtype("float32")
|
|
|
|
|
|
# 清理GPU缓存
|
|
|
if paddle.device.cuda.device_count() > 0:
|
|
|
@@ -430,7 +430,7 @@ if __name__ == "__main__":
|
|
|
"input_dir": "../../OmniDocBench/OpenDataLab___OmniDocBench/images",
|
|
|
"output_dir": "./OmniDocBench_Results_Official",
|
|
|
"pipeline": "PP-StructureV3",
|
|
|
- "device": "gpu:0,1,2,3",
|
|
|
+ "device": "gpu:3",
|
|
|
"instances_per_device": 1,
|
|
|
"batch_size": 1,
|
|
|
# "test_mode": False
|