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

CV4EcologySchool · Aug 4, 2023 · 6abab4d · 6abab4d
1 parent 09a0378
commit 6abab4d
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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()