Skip to content

GitLab

Unverified Commit bd35e9d0 authored by Xiaomeng Zhao's avatar Xiaomeng Zhao Committed by GitHub
Browse files

Merge pull request #1636 from myhloli/dev 

perf(model): adjust batch ratio for GPU memory range
parents bbc8a9a0 29e7a948
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 2 deletions
magic_pdf/model/doc_analyze_by_custom_model.py
View file @ bd35e9d0
......@@ -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
......
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