update doc

This commit is contained in:
Jade Choghari
2025-09-03 11:51:28 -04:00
parent eb564f8ddb
commit 32c1649085
+5 -4
View File
@@ -39,9 +39,9 @@ python src/lerobot/scripts/eval.py \
--eval.n_episodes=3 --eval.n_episodes=3
``` ```
- `-env.task` picks the suite (`libero_object`, `libero_spatial`, etc.). - `--env.task` picks the suite (`libero_object`, `libero_spatial`, etc.).
- `-eval.batch_size` controls how many environments run in parallel. - `--eval.batch_size` controls how many environments run in parallel.
- `-eval.n_episodes` sets how many episodes to run in total. - `--eval.n_episodes` sets how many episodes to run in total.
--- ---
@@ -98,9 +98,10 @@ The environment expects:
```bash ```bash
python src/lerobot/scripts/train.py \ python src/lerobot/scripts/train.py \
--policy.type=smolvla \ --policy.type=smolvla \
--policy.repo_id=${HF_USER}/libero-test \
--dataset.repo_id=jadechoghari/smol-libero3 \ --dataset.repo_id=jadechoghari/smol-libero3 \
--env.type=libero \ --env.type=libero \
--env.task=libero_10,libero_spatial \ --env.task=libero_10 \
--output_dir=./outputs/ \ --output_dir=./outputs/ \
--steps=100000 \ --steps=100000 \
--batch_size=4 \ --batch_size=4 \