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

Update lerobot/scripts/eval.py 

Co-authored-by: Remi <re.cadene@gmail.com>
This commit is contained in:
mshukor 2025-04-02 17:34:05 +02:00 committed by GitHub
parent fedfa60d6e
commit f4d3a70e1f
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
lerobot/scripts/eval.py
View File

@ -156,8 +156,8 @@ def rollout(
key: observation[key].to(device, non_blocking=device.type == "cuda") for key in observation key: observation[key].to(device, non_blocking=device.type == "cuda") for key in observation
} }
# Infer "task" from envs. Works with AsyncVectorEnv. # Infer "task" from attributes of environments. Works with AsyncVectorEnv.
observation = infer_envs_task(env, observation) observation = addd_envs_task(env, observation)
with torch.inference_mode(): with torch.inference_mode():
action = policy.select_action(observation) action = policy.select_action(observation)