Merge pull request #158 from holgerroth/wandb_name

add name to wandb logger
jwohlwend · Jan 24, 2025 · 2ba1f43 · 2ba1f43
2 parents c52e8d5 + fa6f4d0
commit 2ba1f43
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/scripts/train/train.py b/scripts/train/train.py
@@ -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"],