From fa6f4d06b382e2d799642bab09a141f4a93850aa Mon Sep 17 00:00:00 2001
From: Holger Roth <hroth@nvidia.com>
Date: Fri, 17 Jan 2025 16:51:31 -0500
Subject: [PATCH] add name to wandb logger

---
 scripts/train/train.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/scripts/train/train.py b/scripts/train/train.py
index 0185551..7a6491d 100644
--- 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"],