From 91e0e4e175236b859cdc463d8d3418b22d9c2ef8 Mon Sep 17 00:00:00 2001 From: Alexander Soare Date: Mon, 8 Apr 2024 15:05:40 +0100 Subject: [PATCH] rever change --- lerobot/scripts/eval.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/lerobot/scripts/eval.py b/lerobot/scripts/eval.py index 72966211..802a2eb6 100644 --- a/lerobot/scripts/eval.py +++ b/lerobot/scripts/eval.py @@ -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