southfreebird commited on
Commit ·
551fe3d
1
Parent(s): d1357b8
Update README
Browse files
README.md
CHANGED
|
@@ -2,18 +2,18 @@
|
|
| 2 |
datasets: HuggingFaceVLA/libero
|
| 3 |
library_name: lerobot
|
| 4 |
license: apache-2.0
|
| 5 |
-
model_name:
|
| 6 |
pipeline_tag: robotics
|
| 7 |
tags:
|
| 8 |
- lerobot
|
| 9 |
- robotics
|
| 10 |
-
-
|
| 11 |
---
|
| 12 |
|
| 13 |
-
# Model Card for vla0
|
| 14 |
-
|
| 15 |
-
<!-- Provide a quick summary of what the model is/does. -->
|
| 16 |
|
|
|
|
|
|
|
| 17 |
|
| 18 |
_Model type not recognized — please update this template._
|
| 19 |
|
|
@@ -26,31 +26,30 @@ See the full documentation at [LeRobot Docs](https://huggingface.co/docs/lerobot
|
|
| 26 |
## How to Get Started with the Model
|
| 27 |
|
| 28 |
For a complete walkthrough, see the [training guide](https://huggingface.co/docs/lerobot/il_robots#train-a-policy).
|
|
|
|
|
|
|
|
|
|
|
|
|
| 29 |
Below is the short version on how to train and run inference/eval:
|
| 30 |
|
| 31 |
-
### Train from scratch
|
| 32 |
|
| 33 |
```bash
|
| 34 |
-
lerobot-train
|
| 35 |
-
--dataset.repo_id=${HF_USER}/<dataset> \
|
| 36 |
-
--policy.type=act \
|
| 37 |
-
--output_dir=outputs/train/<desired_policy_repo_id> \
|
| 38 |
-
--job_name=lerobot_training \
|
| 39 |
-
--policy.device=cuda \
|
| 40 |
-
--policy.repo_id=${HF_USER}/<desired_policy_repo_id>
|
| 41 |
-
--wandb.enable=true
|
| 42 |
```
|
| 43 |
|
| 44 |
_Writes checkpoints to `outputs/train/<desired_policy_repo_id>/checkpoints/`._
|
| 45 |
|
| 46 |
-
### Evaluate the policy
|
| 47 |
|
| 48 |
```bash
|
| 49 |
-
lerobot-
|
| 50 |
-
--
|
| 51 |
-
--
|
| 52 |
-
--policy.
|
| 53 |
-
--
|
|
|
|
|
|
|
| 54 |
```
|
| 55 |
|
| 56 |
Prefix the dataset repo with **eval\_** and supply `--policy.path` pointing to a local or hub checkpoint.
|
|
|
|
| 2 |
datasets: HuggingFaceVLA/libero
|
| 3 |
library_name: lerobot
|
| 4 |
license: apache-2.0
|
| 5 |
+
model_name: vla0_smol
|
| 6 |
pipeline_tag: robotics
|
| 7 |
tags:
|
| 8 |
- lerobot
|
| 9 |
- robotics
|
| 10 |
+
- vla0_smol
|
| 11 |
---
|
| 12 |
|
| 13 |
+
# Model Card for vla0-smol
|
|
|
|
|
|
|
| 14 |
|
| 15 |
+
Read about the model here: https://robot-learning-collective.github.io/vla-0-smol
|
| 16 |
+
This is VLA-0-Smol trained on the Libero dataset.
|
| 17 |
|
| 18 |
_Model type not recognized — please update this template._
|
| 19 |
|
|
|
|
| 26 |
## How to Get Started with the Model
|
| 27 |
|
| 28 |
For a complete walkthrough, see the [training guide](https://huggingface.co/docs/lerobot/il_robots#train-a-policy).
|
| 29 |
+
|
| 30 |
+
Support of the model is implemented in repo: https://github.com/Robot-Learning-Collective/lerobot-experiments
|
| 31 |
+
|
| 32 |
+
|
| 33 |
Below is the short version on how to train and run inference/eval:
|
| 34 |
|
| 35 |
+
### Train from scratch on PushT
|
| 36 |
|
| 37 |
```bash
|
| 38 |
+
lerobot-train --config_path=configs/vla0_smol.json
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 39 |
```
|
| 40 |
|
| 41 |
_Writes checkpoints to `outputs/train/<desired_policy_repo_id>/checkpoints/`._
|
| 42 |
|
| 43 |
+
### Evaluate the policy on Libero
|
| 44 |
|
| 45 |
```bash
|
| 46 |
+
lerobot-eval \
|
| 47 |
+
--policy.path="Robot-Learning-Collective/VLA-0-Smol" \
|
| 48 |
+
--policy.n_action_steps=0 \
|
| 49 |
+
--policy.ensemble_size=8 \
|
| 50 |
+
--env.type=libero \
|
| 51 |
+
--env.task=libero_object \
|
| 52 |
+
--eval.batch_size=2
|
| 53 |
```
|
| 54 |
|
| 55 |
Prefix the dataset repo with **eval\_** and supply `--policy.path` pointing to a local or hub checkpoint.
|