YOLOv5 updates (#90)

Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>

ultralytics/yolo/utils/callbacks/clearml.py

@@ -23,9 +23,9 @@ def on_train_start(trainer):
 def on_val_end(trainer):
     if trainer.epoch == 0:
         model_info = {
-            "Inference speed (ms/img)": round(trainer.validator.speed[1], 1),
+            "Parameters": get_num_params(trainer.model),
             "GFLOPs": round(get_flops(trainer.model), 1),
-            "Parameters": get_num_params(trainer.model)}
+            "Inference speed (ms/img)": round(trainer.validator.speed[1], 1)}
         Task.current_task().connect(model_info, name='Model')
 
 

ultralytics/yolo/utils/callbacks/tb.py

@@ -11,8 +11,6 @@ def _log_scalars(scalars, step=0):
 def on_train_start(trainer):
     global writer
     writer = SummaryWriter(str(trainer.save_dir))
-    trainer.console.info(f"Logging results to {trainer.save_dir}\n"
-                         f"Starting training for {trainer.args.epochs} epochs...")
 
 
 def on_batch_end(trainer):