Eval reproduction works with gym_aloha

2024-04-08 10:23:26 +01:00
parent e982c732f1
commit 1bab4a1dd5
6 changed files with 66 additions and 103 deletions
--- a/lerobot/common/policies/factory.py
+++ b/lerobot/common/policies/factory.py
@@ -26,7 +26,6 @@ def make_policy(cfg):
        policy = ActionChunkingTransformerPolicy(
            cfg.policy,
            cfg.device,
-            n_obs_steps=cfg.n_obs_steps,
            n_action_steps=cfg.n_action_steps,
        )
    else: