Skip to content

GitLab

"git@developer.sourcefind.cn:modelzoo/resnet50_tensorflow.git" did not exist on "8f23bc72da88528ef84db36c8ee371edabda33d9"
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( ...@@ -182,9 +182,9 @@ def doc_analyze(
batch_ratio = 2 batch_ratio = 2
elif 10 <= gpu_memory <= 12: elif 10 <= gpu_memory <= 12:
batch_ratio = 4 batch_ratio = 4
elif 12 < gpu_memory <= 16: elif 12 < gpu_memory <= 20:
batch_ratio = 8 batch_ratio = 8
elif 16 < gpu_memory <= 24: elif 20 < gpu_memory <= 32:
batch_ratio = 16 batch_ratio = 16
else: else:
batch_ratio = 32 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