From 6abab4d9cab8f2541201a18e5a919c18fa3cbc23 Mon Sep 17 00:00:00 2001 From: David de la Iglesia Castro Date: Fri, 4 Aug 2023 19:39:36 +0200 Subject: [PATCH] Mark DVC `trainer.best` artifact as `type=model` (#4143) --- ultralytics/utils/callbacks/dvc.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/ultralytics/utils/callbacks/dvc.py b/ultralytics/utils/callbacks/dvc.py index c81c492a..ef4c0cce 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()