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)