From 26d9a070d80af25fdbef2e02c722574450ad1e36 Mon Sep 17 00:00:00 2001 From: Alexander Soare Date: Tue, 7 May 2024 09:09:21 +0100 Subject: [PATCH] Add policy and dataset tags to WandB logging (#144) --- lerobot/common/logger.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/lerobot/common/logger.py b/lerobot/common/logger.py index 832eaa7e8..8e7fe7f2d 100644 --- a/lerobot/common/logger.py +++ b/lerobot/common/logger.py @@ -17,9 +17,10 @@ def log_output_dir(out_dir): def cfg_to_group(cfg, return_list=False): - """Return a wandb-safe group name for logging. Optionally returns group name as list.""" - # lst = [cfg.task, cfg.modality, re.sub("[^0-9a-zA-Z]+", "-", cfg.exp_name)] + """Return a group name for logging. Optionally returns group name as list.""" lst = [ + f"policy:{cfg.policy.name}", + f"dataset:{cfg.dataset_repo_id}", f"env:{cfg.env.name}", f"seed:{cfg.seed}", ]