From 5d4af4b0b1fc1f77c0151747f1a9a7de20a2aff8 Mon Sep 17 00:00:00 2001 From: yfynb1111 <33393037+yfynb1111@users.noreply.github.com> Date: Fri, 17 Oct 2025 22:32:56 +0800 Subject: [PATCH] Fix: debug policy load pretrained model failure problem (#2073) Co-authored-by: Steven Palma --- src/lerobot/rl/eval_policy.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/lerobot/rl/eval_policy.py b/src/lerobot/rl/eval_policy.py index 7cec6680..16bb64a7 100644 --- a/src/lerobot/rl/eval_policy.py +++ b/src/lerobot/rl/eval_policy.py @@ -65,7 +65,7 @@ def main(cfg: TrainRLServerPipelineConfig): # env_cfg=cfg.env, 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() eval_policy(env, policy=policy, n_episodes=10)