Add option to disable tags on WandB (#1339)

Signed-off-by: Steven Palma <imstevenpmwork@ieee.org>
Co-authored-by: Steven Palma <imstevenpmwork@ieee.org>
This commit is contained in:
Bruno Machado
2026-03-12 00:54:08 +01:00
committed by GitHub
parent efee611403
commit 0db5f66dda
2 changed files with 2 additions and 1 deletions
+1
View File
@@ -47,6 +47,7 @@ class WandBConfig:
notes: str | None = None
run_id: str | None = None
mode: str | None = None # Allowed values: 'online', 'offline' 'disabled'. Defaults to 'online'
add_tags: bool = True # If True, save configuration as tags in the WandB run.
@dataclass
+1 -1
View File
@@ -98,7 +98,7 @@ class WandBLogger:
entity=self.cfg.entity,
name=self.job_name,
notes=self.cfg.notes,
tags=cfg_to_group(cfg, return_list=True, truncate_tags=True),
tags=cfg_to_group(cfg, return_list=True, truncate_tags=True) if self.cfg.add_tags else None,
dir=self.log_dir,
config=cfg.to_dict(),
# TODO(rcadene): try set to True