From 60e42e16c2bc51b303e680afccd72351af7a7a69 Mon Sep 17 00:00:00 2001
From: Glenn Jocher <glenn.jocher@ultralytics.com>
Date: Sat, 6 Nov 2021 12:21:17 +0100
Subject: [PATCH] Update autobatch.py (#5536)

---
 utils/autobatch.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/utils/autobatch.py b/utils/autobatch.py
index 168b16f..1632e9b 100644
--- a/utils/autobatch.py
+++ b/utils/autobatch.py
@@ -52,5 +52,5 @@ def autobatch(model, imgsz=640, fraction=0.9, batch_size=16):
     batch_sizes = batch_sizes[:len(y)]
     p = np.polyfit(batch_sizes, y, deg=1)  # first degree polynomial fit
     b = int((f * fraction - p[1]) / p[0])  # y intercept (optimal batch size)
-    print(f'{prefix}Using colorstr(batch-size {b}) for {d} {t * fraction:.3g}G/{t:.3g}G ({fraction * 100:.0f}%)')
+    print(f'{prefix}Using batch-size {b} for {d} {t * fraction:.3g}G/{t:.3g}G ({fraction * 100:.0f}%)')
     return b