wangzixiang
/
lerobot
mirror of https://github.com/huggingface/lerobot.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Sanitize cfg.wandb

This commit is contained in:
Cadene 2024-02-25 11:15:09 +00:00
parent 5812298290
commit 0eb9b5d1a5
2 changed files with 10 additions and 11 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
lerobot/common/logger.py
View File

@ -137,11 +137,10 @@ class Logger(object):
self._cfg = cfg self._cfg = cfg
self._eval = [] self._eval = []
print_run(cfg) print_run(cfg)
project = cfg.get("wandb_project", "none") project = cfg.get("wandb", {}).get("project")
entity = cfg.get("wandb_entity", "none") entity = cfg.get("wandb", {}).get("entity")
run_offline = ( enable_wandb = cfg.get("wandb", {}).get("enable", False)
not cfg.get("use_wandb", False) or project == "none" or entity == "none" run_offline = not enable_wandb or not project or not entity
)
if run_offline: if run_offline:
print(colored("Logs will be saved locally.", "yellow", attrs=["bold"])) print(colored("Logs will be saved locally.", "yellow", attrs=["bold"]))
self._wandb = None self._wandb = None
@ -153,7 +152,7 @@ class Logger(object):
project=project, project=project,
entity=entity, entity=entity,
name=job_name, name=job_name,
notes=cfg.notes, notes=cfg.get("wandb", {}).get("notes"),
# group=self._group, # group=self._group,
tags=cfg_to_group(cfg, return_list=True), tags=cfg_to_group(cfg, return_list=True),
dir=self._log_dir, dir=self._log_dir,

10
lerobot/configs/default.yaml
View File

@ -107,8 +107,8 @@ policy:
mlp_dim: 512 mlp_dim: 512
latent_dim: 50 latent_dim: 50
# wandb wandb:
use_wandb: true enable: true
wandb_project: lerobot project: lerobot
wandb_entity: rcadene # insert your own entity: rcadene # insert your own
notes: "" notes: ""