Parcourir la source

fix: 调整默认批处理大小,从4改为2,以优化内存使用

zhch158_admin il y a 3 mois
Parent
commit
792cdd5879
1 fichiers modifiés avec 1 ajouts et 1 suppressions
  1. 1 1
      zhch/ppstructurev3_single_process.py

+ 1 - 1
zhch/ppstructurev3_single_process.py

@@ -334,7 +334,7 @@ if __name__ == "__main__":
             "output_dir": "./OmniDocBench_Results_Single",
             "pipeline": "PP-StructureV3",
             "device": "gpu:0",
-            "batch_size": 4,
+            "batch_size": 2,
         }
         
         # 构造参数