diff --git a/ultralytics/utils/callbacks/dvc.py b/ultralytics/utils/callbacks/dvc.py index c81c492..ef4c0cc 100644 --- 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()