From f1708c8a37263050d023e90a0234e71646e969fa Mon Sep 17 00:00:00 2001 From: Simon Alibert Date: Wed, 28 Feb 2024 12:35:49 +0100 Subject: [PATCH] install fix --- lerobot/configs/env/pusht.yaml | 1 + lerobot/scripts/eval.py | 2 +- 2 files changed, 2 insertions(+), 1 deletion(-) 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,