Fix W&B project name separator compatibility (#17627)
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com> Co-authored-by: UltralyticsAssistant <web@ultralytics.com>
This commit is contained in:
parent
dc9b99214f
commit
f3241eb676
1 changed files with 6 additions and 1 deletions
|
|
@ -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):
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue