"torchvision/io/video_reader.py" did not exist on "a8d2227fd2542df7643ba204a5304bd2c0ff8deb"
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( ...@@ -184,7 +184,7 @@ def doc_analyze(
batch_ratio = 4 batch_ratio = 4
elif 12 < gpu_memory <= 16: elif 12 < gpu_memory <= 16:
batch_ratio = 8 batch_ratio = 8
elif 16 < gpu_memory <= 24: elif 16 < 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