hangX/lerobot
1
0
Fork 0
forked from tangger/lerobot
Code Pull Requests Activity

only try to start env if it is closed

Browse Source
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
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():