TekbotRobotics/Cotonou_VLA_flags_sorting

SmolVLA: A vision-language-action model for affordable and efficient robotics

Resources and technical documentation:

Train using Google Colab Notebook

SmolVLA HF Documentation

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