فهرست منبع

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

myhloli 4 ماه پیش
والد
کامیت
399023f580
1فایلهای تغییر یافته به همراه2 افزوده شده و 2 حذف شده
  1. 2 2
      mineru/backend/pipeline/pipeline_analyze.py

+ 2 - 2
mineru/backend/pipeline/pipeline_analyze.py

@@ -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)