YOLOv5 updates (#90)

Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
2022-12-25 14:33:18 +01:00
parent ebd3cfb2fd
commit 98815d560f
27 changed files with 281 additions and 161 deletions
--- a/ultralytics/yolo/utils/callbacks/clearml.py
+++ b/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')