Make sure targets are normalized too (#106)

2024-04-26 11:18:39 +01:00
parent b980c5dd9e
commit 45f351c618
8 changed files with 116 additions and 92 deletions
--- a/lerobot/scripts/train.py
+++ b/lerobot/scripts/train.py
@@ -258,7 +258,6 @@ def train(cfg: dict, out_dir=None, job_name=None):
                policy,
                video_dir=Path(out_dir) / "eval",
                max_episodes_rendered=4,
-                transform=offline_dataset.transform,
                seed=cfg.seed,
            )
            log_eval_info(logger, eval_info["aggregated"], step, cfg, offline_dataset, is_offline)