- Update batch ratio calculation for GPU memory range - Increase upper limit for batch ratio 16 from 24 to 32 GB
@@ -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