feat: simplify batch ratio calculation by removing redundant condition for GPU memory

2026-09-21 12:42:22 +08:00 · 2026-06-10 16:28:03 +08:00
parent ac9ac14bbe
commit ffff1f656e
1 changed files with 0 additions and 2 deletions
@@ -882,8 +882,6 @@ def get_batch_ratio(device):
        batch_ratio = 8
    elif gpu_memory >= 12:
        batch_ratio = 4
-    elif gpu_memory >= 8:
-        batch_ratio = 2
    else:
        batch_ratio = 1