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

only try to start env if it is closed

This commit is contained in:
Alexander Soare 2024-03-22 15:32:55 +00:00
parent e698d38a35
commit d43fa600a0
1 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
lerobot/scripts/eval.py
View File

@ -90,6 +90,7 @@ def eval_policy(
if policy is not None: if policy is not None:
policy.clear_action_queue() policy.clear_action_queue()
if env.is_closed:
env.start() # needed to be able to get the seeds the first time as BatchedEnvs are lazy env.start() # needed to be able to get the seeds the first time as BatchedEnvs are lazy
seeds.extend(env._next_seed) seeds.extend(env._next_seed)
with torch.inference_mode(): with torch.inference_mode():