Skip to content

Commit

Permalink
add name to wandb logger
Browse files Browse the repository at this point in the history
  • Loading branch information
holgerroth committed Jan 17, 2025
1 parent c52e8d5 commit fa6f4d0
Showing 1 changed file with 1 addition and 0 deletions.
1 change: 1 addition & 0 deletions scripts/train/train.py
Original file line number Diff line number Diff line change
Expand Up @@ -178,6 +178,7 @@ def train(raw_config: str, args: list[str]) -> None: # noqa: C901, PLR0912, PLR
loggers = []
if wandb:
wdb_logger = WandbLogger(
name=wandb["name"],
group=wandb["name"],
save_dir=cfg.output,
project=wandb["project"],
Expand Down

0 comments on commit fa6f4d0

Please sign in to comment.