Эх сурвалжийг харах

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

zhch158_admin 3 сар өмнө
parent
commit
792cdd5879

+ 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,
         }
         
         # 构造参数