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 = {