metadata
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
Designed by Tekbot Robotics and Inspired from Hugging Face.
This model was finetuned on hugging Face base model.
Before proceeding to the next steps, you need to properly install the environment by following Installation Guide on the docs.
Install smolvla extra dependencies:
pip install -e ".[smolvla]"
Example of finetuning the smolvla pretrained model (smolvla_base):
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