tangger/lerobot
1
0
Fork 1
Code Issues Pull Requests Actions 1 Packages Projects Releases Wiki Activity

fix eval-related doc errors (#2183)

Browse Source 
Signed-off-by: Steven Palma <imstevenpmwork@ieee.org>
Co-authored-by: Steven Palma <imstevenpmwork@ieee.org>
This commit is contained in:
Infinity4B
2025-10-17 20:34:21 +08:00
committed by GitHub
parent 4afb253825
commit 52455d03a7
2 changed files with 3 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
README.md
View File

@@ -310,7 +310,7 @@ To upload these to the hub, run the following:
huggingface-cli upload ${hf_user}/${repo_name} path/to/pretrained_model 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 ### Acknowledgment

4
src/lerobot/scripts/lerobot_eval.py
View File

@@ -27,7 +27,7 @@ lerobot-eval \
--eval.batch_size=10 \ --eval.batch_size=10 \
--eval.n_episodes=10 \ --eval.n_episodes=10 \
--use_amp=false \ --use_amp=false \
--device=cuda --policy.device=cuda
``` ```
OR, you want to evaluate a model checkpoint from the LeRobot training script for 10 episodes. 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.batch_size=10 \
--eval.n_episodes=10 \ --eval.n_episodes=10 \
--use_amp=false \ --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. Note that in both examples, the repo/folder should contain at least `config.json` and `model.safetensors` files.