File size: 1,299 Bytes
f451f66 4bbe671 f451f66 4bbe671 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 |
---
pipeline_tag: robotics
tags:
- lerobot
library_name: lerobot
datasets:
- TekbotRobotics/svla_so101_pickplace_flags_sorting
---
## SmolVLA: A vision-language-action model for affordable and efficient robotics
Resources and technical documentation:
[Train using Google Colab Notebook](https://colab.research.google.com/github/huggingface/notebooks/blob/main/lerobot/training-smolvla.ipynb#scrollTo=ZO52lcQtxseE)
[SmolVLA HF Documentation](https://huggingface.co/docs/lerobot/smolvla)
Designed by Tekbot Robotics and Inspired from Hugging Face.
This model was finetuned on [hugging Face base model](https://huggingface.co/lerobot/smolvla_base/).
Before proceeding to the next steps, you need to properly install the environment by following [Installation Guide](https://huggingface.co/docs/lerobot/installation) on the docs.
Install smolvla extra dependencies:
```bash
pip install -e ".[smolvla]"
```
Example of finetuning the smolvla pretrained model (`smolvla_base`):
```bash
python lerobot/scripts/train.py \
--policy.path=lerobot/smolvla_base \
--dataset.repo_id=TekbotRobotics/svla_so101_pickplace_flags_sorting \
--batch_size=8 \
--steps=2000 \
--output_dir=outputs/train/my_smolvla \
--job_name=my_smolvla_training \
--policy.device=cuda \
--wandb.enable=true
```
|