Commit d1af4566 authored by myhloli's avatar myhloli
Browse files

perf(model): adjust batch ratio for GPU memory range

- Update batch ratio calculation for GPU memory range
- Increase upper limit for batch ratio 16 from 24 to 32 GB
parent bbc8a9a0
......@@ -184,7 +184,7 @@ def doc_analyze(
batch_ratio = 4
elif 12 < gpu_memory <= 16:
batch_ratio = 8
elif 16 < gpu_memory <= 24:
elif 16 < 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