wangzixiang
/
lerobot
mirror of https://github.com/huggingface/lerobot.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

install fix

This commit is contained in:
Simon Alibert 2024-02-28 12:35:49 +01:00
parent 98f8869743
commit f1708c8a37
2 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
lerobot/configs/env/pusht.yaml vendored
View File

@ -18,6 +18,7 @@ env:
image_size: 96 image_size: 96
action_repeat: 1 action_repeat: 1
episode_length: 300 episode_length: 300
fps: ${fps}
policy: policy:
state_dim: 2 state_dim: 2

2
lerobot/scripts/eval.py
View File

@ -109,7 +109,7 @@ def eval(cfg: dict, out_dir=None):
env = make_env(cfg) env = make_env(cfg)
if cfg.pretrained_model_path: if cfg.policy.pretrained_model_path:
policy = make_policy(cfg) policy = make_policy(cfg)
policy = TensorDictModule( policy = TensorDictModule(
policy, policy,