ydy0615/lerobot-clone
1
0
Fork 0
mirror of https://github.com/huggingface/lerobot.git synced 2026-06-03 20:31:25 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix(scripts): add missing observation overwrite in eval and async (#2265)

Browse Source
This commit is contained in:
Steven Palma
2025-10-20 23:34:24 +02:00
committed by GitHub
parent 5f6f476f32
commit b954337ac7
4 changed files with 12 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
src/lerobot/scripts/lerobot_eval.py
View File

@@ -508,7 +508,10 @@ def eval_main(cfg: EvalPipelineConfig):
policy_cfg=cfg.policy,
pretrained_path=cfg.policy.pretrained_path,
# The inference device is automatically set to match the detected hardware, overriding any previous device settings from training to ensure compatibility.
preprocessor_overrides={"device_processor": {"device": str(policy.config.device)}},
preprocessor_overrides={
"device_processor": {"device": str(policy.config.device)},
"rename_observations_processor": {"rename_map": cfg.rename_map},
},
)
with torch.no_grad(), torch.autocast(device_type=device.type) if cfg.policy.use_amp else nullcontext():
info = eval_policy_all(