From f3241eb67674e6d76a2dd2b914b242770590b083 Mon Sep 17 00:00:00 2001 From: Kyre Date: Tue, 19 Nov 2024 18:55:58 +0800 Subject: [PATCH] Fix W&B project name separator compatibility (#17627) Co-authored-by: Glenn Jocher Co-authored-by: UltralyticsAssistant --- ultralytics/utils/callbacks/wb.py | 7 ++++++- 1 file changed, 6 insertions(+), 1 deletion(-) diff --git a/ultralytics/utils/callbacks/wb.py b/ultralytics/utils/callbacks/wb.py index 22bbc347..0898a293 100644 --- a/ultralytics/utils/callbacks/wb.py +++ b/ultralytics/utils/callbacks/wb.py @@ -109,7 +109,12 @@ def _log_plots(plots, step): def on_pretrain_routine_start(trainer): """Initiate and start project if module is present.""" - wb.run or wb.init(project=trainer.args.project or "Ultralytics", name=trainer.args.name, config=vars(trainer.args)) + if not wb.run: + wb.init( + project=str(trainer.args.project).replace("/", "-") if trainer.args.project else "Ultralytics", + name=str(trainer.args.name).replace("/", "-"), + config=vars(trainer.args), + ) def on_fit_epoch_end(trainer):