Commit 399023f5 authored by myhloli's avatar myhloli
Browse files

fix: update default value of MINERU_MIN_BATCH_INFERENCE_SIZE to 128 in pipeline_analyze.py

parent b420094b
......@@ -75,9 +75,9 @@ def doc_analyze(
):
"""
适当调大MIN_BATCH_INFERENCE_SIZE可以提高性能,可能会增加显存使用量,
可通过环境变量MINERU_MIN_BATCH_INFERENCE_SIZE设置,默认值为100
可通过环境变量MINERU_MIN_BATCH_INFERENCE_SIZE设置,默认值为128
"""
min_batch_inference_size = int(os.environ.get('MINERU_MIN_BATCH_INFERENCE_SIZE', 100))
min_batch_inference_size = int(os.environ.get('MINERU_MIN_BATCH_INFERENCE_SIZE', 128))
# 收集所有页面信息
all_pages_info = [] # 存储(dataset_index, page_index, img, ocr, lang, width, height)
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment