Mark DVC `trainer.best` artifact as `type=model` (#4143)

2 years ago · 6abab4d9ca
parent 09a0378e81
commit 6abab4d9ca
1 changed files with 1 additions and 1 deletions
--- a/ultralytics/utils/callbacks/dvc.py
+++ b/ultralytics/utils/callbacks/dvc.py
@ -123,7 +123,7 @@ def on_train_end(trainer):
        _log_confusion_matrix(trainer.validator)

        if trainer.best.exists():
-            live.log_artifact(trainer.best, copy=True)
+            live.log_artifact(trainer.best, copy=True, type='model')

        live.end()