- 将 batch_size 参数从 4 修改为 1 - 此修改可能会影响模型的性能和资源利用率
@@ -432,7 +432,7 @@ if __name__ == "__main__":
"pipeline": "PP-StructureV3",
"device": "gpu:0,1,2,3",
"instances_per_device": 1,
- "batch_size": 4,
+ "batch_size": 1,
# "test_mode": False
}