Update

ultralytics · Oct 23, 2021 · 7537daa · 7537daa
1 parent e8ccd55
commit 7537daa
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/utils/autobatch.py b/utils/autobatch.py
@@ -19,7 +19,7 @@ def check_batch_size(model, imgsz=640, b=16):
     return b
 
 
-def autobatch(model, imgsz=640, fraction=0.95):
+def autobatch(model, imgsz=640, fraction=0.9):
     # Automatically compute optimal batch size to use `fraction` of available CUDA memory
     # Usage:
     #     import torch