Provide more information to the user (#358)

Co-authored-by: Alexander Soare <alexander.soare159@gmail.com> Co-authored-by: Remi <re.cadene@gmail.com>
2024-08-23 18:00:35 +08:00
parent b5ad79a7d3
commit a2592a5563
8 changed files with 76 additions and 15 deletions
--- a/lerobot/scripts/train.py
+++ b/lerobot/scripts/train.py
@@ -241,6 +241,7 @@ def train(cfg: DictConfig, out_dir: str | None = None, job_name: str | None = No
        raise NotImplementedError()

    init_logging()
+    logging.info(pformat(OmegaConf.to_container(cfg)))

    if cfg.training.online_steps > 0 and isinstance(cfg.dataset_repo_id, ListConfig):
        raise NotImplementedError("Online training with LeRobotMultiDataset is not implemented.")