Forráskód Böngészése

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

zhch158_admin 3 hónapja
szülő
commit
792cdd5879
1 módosított fájl, 1 hozzáadás és 1 törlés
  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,
         }
         
         # 构造参数