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

Co-authored-by: Steven Palma <imstevenpmwork@ieee.org>
2025-10-17 22:32:56 +08:00
parent 0050d7c61c
commit 5d4af4b0b1
1 changed files with 1 additions and 1 deletions
--- 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)