install fix

2024-02-28 12:35:49 +01:00 · 2024-02-28 12:35:49 +01:00 · f1708c8a37
parent 98f8869743
commit f1708c8a37
2 changed files with 2 additions and 1 deletions
--- 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
--- 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,