Răsfoiți Sursa

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

- 将 batch_size 参数从 4 修改为 1
- 此修改可能会影响模型的性能和资源利用率
zhch158_admin 3 luni în urmă
părinte
comite
e361db0d3c
1 a modificat fișierele cu 1 adăugiri și 1 ștergeri
  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
         }