admin/lerobot
1
0
Fork 0
mirror of https://github.com/huggingface/lerobot.git synced 2026-05-23 04:30:10 +00:00
Code Issues Packages Projects Releases Wiki Activity

Add option to disable tags on WandB (#1339)

Browse Source 
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
Download Patch File Download Diff File Expand all files Collapse all files
src/lerobot/configs/default.py
+1
View File
@@ -47,6 +47,7 @@ class WandBConfig:
notes: str | None = None notes: str | None = None
run_id: str | None = None run_id: str | None = None
mode: str | None = None # Allowed values: 'online', 'offline' 'disabled'. Defaults to 'online' 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 @dataclass
src/lerobot/rl/wandb_utils.py
+1 -1
View File
@@ -98,7 +98,7 @@ class WandBLogger:
entity=self.cfg.entity, entity=self.cfg.entity,
name=self.job_name, name=self.job_name,
notes=self.cfg.notes, 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, dir=self.log_dir,
config=cfg.to_dict(), config=cfg.to_dict(),
# TODO(rcadene): try set to True # TODO(rcadene): try set to True