Преглед изворни кода

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

zhch158_admin пре 3 месеци
родитељ
комит
792cdd5879
1 измењених фајлова са 1 додато и 1 уклоњено
  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,
         }
         
         # 构造参数