Add gym-aloha, rename simxarm -> xarm, refactor

2024-04-08 16:18:53 +02:00
parent 5dff6d8339
commit 3f6dfa4916
15 changed files with 91 additions and 97 deletions
--- a/lerobot/scripts/train.py
+++ b/lerobot/scripts/train.py
@@ -162,7 +162,7 @@ def train(cfg: dict, out_dir=None, job_name=None):
    logger = Logger(out_dir, job_name, cfg)

    log_output_dir(out_dir)
-    logging.info(f"{cfg.env.task=}")
+    logging.info(f"{cfg.env.handle=}")
    logging.info(f"{cfg.offline_steps=} ({format_big_number(cfg.offline_steps)})")
    logging.info(f"{cfg.online_steps=}")
    logging.info(f"{cfg.env.action_repeat=}")