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.