admin/lerobot
1
0
Fork 0
mirror of https://github.com/huggingface/lerobot.git synced 2026-05-22 03:59:42 +00:00
Code Issues Packages Projects Releases Wiki Activity

Fix: debug policy load pretrained model failure problem (#2073)

Browse Source 
Co-authored-by: Steven Palma <imstevenpmwork@ieee.org>
This commit is contained in:
yfynb1111
2025-10-17 22:32:56 +08:00
committed by GitHub
parent 0050d7c61c
commit 5d4af4b0b1
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/lerobot/rl/eval_policy.py
+1 -1
View File
@@ -65,7 +65,7 @@ def main(cfg: TrainRLServerPipelineConfig):
# env_cfg=cfg.env, # env_cfg=cfg.env,
ds_meta=dataset_meta, ds_meta=dataset_meta,
) )
policy.from_pretrained(env_cfg.pretrained_policy_name_or_path) policy = policy.from_pretrained(env_cfg.pretrained_policy_name_or_path)
policy.eval() policy.eval()
eval_policy(env, policy=policy, n_episodes=10) eval_policy(env, policy=policy, n_episodes=10)