浏览代码

refactor(zhch): 修改 PP-StructureV3 多 GPU 多进程脚本的批量大小

- 将 batch_size 参数从 4 修改为 1
- 此修改可能会影响模型的性能和资源利用率
zhch158_admin 3 月之前
父节点
当前提交
e361db0d3c
共有 1 个文件被更改,包括 1 次插入1 次删除
  1. 1 1
      zhch/ppstructurev3_multi_gpu_multiprocess_official.py

+ 1 - 1
zhch/ppstructurev3_multi_gpu_multiprocess_official.py

@@ -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
         }