diff --git a/lerobot/configs/env/pusht.yaml b/lerobot/configs/env/pusht.yaml index fd0f70c1..1719321b 100644 --- a/lerobot/configs/env/pusht.yaml +++ b/lerobot/configs/env/pusht.yaml @@ -18,6 +18,7 @@ env: image_size: 96 action_repeat: 1 episode_length: 300 + fps: ${fps} policy: state_dim: 2 diff --git a/lerobot/scripts/eval.py b/lerobot/scripts/eval.py index eaade622..2e8f6965 100644 --- a/lerobot/scripts/eval.py +++ b/lerobot/scripts/eval.py @@ -109,7 +109,7 @@ def eval(cfg: dict, out_dir=None): env = make_env(cfg) - if cfg.pretrained_model_path: + if cfg.policy.pretrained_model_path: policy = make_policy(cfg) policy = TensorDictModule( policy,