Przeglądaj źródła

Merge pull request #1636 from myhloli/dev

perf(model): adjust batch ratio for GPU memory range
Xiaomeng Zhao 9 miesięcy temu
rodzic
commit
bd35e9d013
1 zmienionych plików z 2 dodań i 2 usunięć
  1. 2 2
      magic_pdf/model/doc_analyze_by_custom_model.py

+ 2 - 2
magic_pdf/model/doc_analyze_by_custom_model.py

@@ -182,9 +182,9 @@ def doc_analyze(
                 batch_ratio = 2
             elif 10 <= gpu_memory <= 12:
                 batch_ratio = 4
-            elif 12 < gpu_memory <= 16:
+            elif 12 < gpu_memory <= 20:
                 batch_ratio = 8
-            elif 16 < gpu_memory <= 24:
+            elif 20 < gpu_memory <= 32:
                 batch_ratio = 16
             else:
                 batch_ratio = 32