Update yolo.py

ultralytics · Apr 26, 2021 · c17e501 · c17e501
1 parent 0c1e90a
commit c17e501
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/models/yolo.py b/models/yolo.py
@@ -136,7 +136,7 @@ def forward_once(self, x, profile=False):
                 dt.append((time_synchronized() - t) * 100)
                 if m == self.model[0]:
                     logger.info(f"{'time (ms)':>10s} {'GFLOPS':>10s} {'params':>10s}  {'module'}")
-                logger.info(f'{dt[-1]:10.1f} {o:10.1f} {m.np:10.0f}  {m.type}')
+                logger.info(f'{dt[-1]:10.2f} {o:10.2f} {m.np:10.0f}  {m.type}')
 
             x = m(x)  # run
             y.append(x if m.i in self.save else None)  # save output