From 52455d03a734bed9fd481ad3511db1379c55a653 Mon Sep 17 00:00:00 2001 From: Infinity4B <54246104+Infinity4B@users.noreply.github.com> Date: Fri, 17 Oct 2025 20:34:21 +0800 Subject: [PATCH] fix eval-related doc errors (#2183) Signed-off-by: Steven Palma Co-authored-by: Steven Palma --- README.md | 2 +- src/lerobot/scripts/lerobot_eval.py | 4 ++-- 2 files changed, 3 insertions(+), 3 deletions(-) diff --git a/README.md b/README.md index 357e62cc..6409d931 100644 --- a/README.md +++ b/README.md @@ -310,7 +310,7 @@ To upload these to the hub, run the following: huggingface-cli upload ${hf_user}/${repo_name} path/to/pretrained_model ``` -See [eval.py](https://github.com/huggingface/lerobot/blob/main/src/lerobot/scripts/eval.py) for an example of how other people may use your policy. +See [lerobot_eval.py](https://github.com/huggingface/lerobot/blob/main/src/lerobot/scripts/lerobot_eval.py) for an example of how other people may use your policy. ### Acknowledgment diff --git a/src/lerobot/scripts/lerobot_eval.py b/src/lerobot/scripts/lerobot_eval.py index aed7d32e..0fdec928 100644 --- a/src/lerobot/scripts/lerobot_eval.py +++ b/src/lerobot/scripts/lerobot_eval.py @@ -27,7 +27,7 @@ lerobot-eval \ --eval.batch_size=10 \ --eval.n_episodes=10 \ --use_amp=false \ - --device=cuda + --policy.device=cuda ``` OR, you want to evaluate a model checkpoint from the LeRobot training script for 10 episodes. @@ -38,7 +38,7 @@ lerobot-eval \ --eval.batch_size=10 \ --eval.n_episodes=10 \ --use_amp=false \ - --device=cuda + --policy.device=cuda ``` Note that in both examples, the repo/folder should contain at least `config.json` and `model.safetensors` files.