From 3821d4a72c9b435b51a1ad3ae641ed809c0c05a6 Mon Sep 17 00:00:00 2001
From: Soumik Rakshit <19soumik.rakshit96@gmail.com>
Date: Thu, 6 Jul 2023 17:53:55 +0530
Subject: [PATCH] Add alias `"best"` for the model checkpoint artifact (#3558)

Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
---
 ultralytics/yolo/utils/callbacks/wb.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/ultralytics/yolo/utils/callbacks/wb.py b/ultralytics/yolo/utils/callbacks/wb.py
index 827f797..4b4c29b 100644
--- a/ultralytics/yolo/utils/callbacks/wb.py
+++ b/ultralytics/yolo/utils/callbacks/wb.py
@@ -50,7 +50,7 @@ def on_train_end(trainer):
     art = wb.Artifact(type='model', name=f'run_{wb.run.id}_model')
     if trainer.best.exists():
         art.add_file(trainer.best)
-        wb.run.log_artifact(art)
+        wb.run.log_artifact(art, aliases=['best'])
 
 
 callbacks = {