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

rever change

This commit is contained in:
Alexander Soare 2024-04-08 15:05:40 +01:00
parent 0b4c42f4ff
commit 91e0e4e175
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
lerobot/scripts/eval.py
View File

@ -255,7 +255,7 @@ def eval(cfg: dict, out_dir=None, stats_path=None):
transform = make_dataset(cfg, stats_path=stats_path).transform
logging.info("Making environment.")
env = make_env(cfg, num_parallel_envs=cfg.rollout_batch_size)
env = make_env(cfg, num_parallel_envs=cfg.eval_episodes)
# when policy is None, rollout a random policy
policy = make_policy(cfg) if cfg.policy.pretrained_model_path else None