Update

ultralytics · Oct 8, 2021 · 94f59b5 · 94f59b5
1 parent d4c98a7
commit 94f59b5
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/utils/autobatch.py b/utils/autobatch.py
@@ -36,7 +36,7 @@ def autobatch(model, imgsz=640, fraction=0.9):
     f = t - (r + a)  # free inside reserved
     print(f'{prefix}{t:.3g}G total, {r:.3g}G reserved, {a:.3g}G allocated, {f:.3g}G free')
 
-    batch_sizes = [1, 2, 4, 8, 16, 32]
+    batch_sizes = [1, 2, 4, 8, 16]
     try:
         img = [torch.zeros(b, 3, imgsz, imgsz) for b in batch_sizes]
         y = profile(img, model, n=3, device=device)