소스 검색

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
         }