Robotics
LeRobot
Safetensors
smolvla

Model Card for smolvla

SmolVLA is a compact, efficient vision-language-action model that achieves competitive performance at reduced computational costs and can be deployed on consumer-grade hardware.

smolvla architecture

This policy has been trained and pushed to the Hub using LeRobot.

Learn how to train and run it in the LeRobot smolvla guide, or browse the full documentation.


Model Details

  • License: apache-2.0
  • Fine-tuned from: lerobot/smolvla_base
  • Robot type: so_follower
  • Cameras: front, second

Inputs & Outputs

The policy consumes these observation features and produces these action features.

Inputs

Feature Type Shape
observation.state STATE (6,)
observation.images.camera1 VISUAL (3, 256, 256)
observation.images.camera2 VISUAL (3, 256, 256)
observation.images.camera3 VISUAL (3, 256, 256)

Outputs

Feature Type Shape
action ACTION (6,)

Training Dataset

  • Repository: merged
  • Episodes: 237
  • Frames: 118666
  • Frame rate: 30 FPS
  • Task(s): "putting green cube from red plate to green plate", "putting red cube from red plate to green plate", "putting gray cube from red plate to green plate", "putting green cube from green plate to red plate", "putting red cube from green plate to red plate", "putting gray cube from green plate to red plate", "red cubes from red plate to green plate but opposite sides", "gray cubes from red plate to green plate but opposite sides", "green cubes from red plate to green plate but opposite sides", "green cubes from green plate to red plate but opposite sides", "red cubes from green plate to red plate but opposite sides", "GRAY cubes from green plate to red plate but opposite sides"

Training Configuration

Setting Value
Training steps 20000
Batch size 4
Optimizer adamw
Learning rate 0.0001
Seed 1000
LeRobot version 0.6.2

How to Get Started with the Model

New to LeRobot? These guides cover the full workflow:

The short version to run and train this policy:

Run the policy on your robot

lerobot-rollout \
  --strategy.type=base \
  --robot.type=so_follower \
  --robot.port=<your_robot_port> \
  --robot.cameras="{ <camera_1>: {type: opencv, index_or_path: <index_or_path>, width: 640, height: 480, fps: 30}, <camera_2>: {type: opencv, index_or_path: <index_or_path>, width: 640, height: 480, fps: 30}}" \
  --policy.path=smolvla_merged \
  --task="putting green cube from red plate to green plate" \
  --duration=60

Replace the remaining <...> placeholders with your own values: --robot.port and the camera names/indices are specific to your machine, and the camera names must match the observation keys this policy was trained on.

When --strategy.type=base is used the script doesn't record the episodes. Skipping duration will make the policy run indefinitely. For more information look at rollout documentation.

Train your own policy

This policy type is usually fine-tuned from the pretrained base model lerobot/smolvla_base:

lerobot-train \
  --dataset.repo_id=${HF_USER}/<dataset> \
  --policy.path=lerobot/smolvla_base \
  --output_dir=outputs/train/<policy_repo_id> \
  --job_name=lerobot_training \
  --policy.device=cuda \
  --policy.repo_id=${HF_USER}/<policy_repo_id> \
  --wandb.enable=true

Writes checkpoints to outputs/train/<policy_repo_id>/checkpoints/.


Evaluation

No evaluation results have been provided for this policy yet.


Citation

If you use this policy, please cite the method linked in the description above, along with LeRobot:

@misc{cadene2024lerobot,
    author = {Cadene, Remi and Alibert, Simon and Soare, Alexander and Gallouedec, Quentin and Zouitine, Adil and Palma, Steven and Kooijmans, Pepijn and Aractingi, Michel and Shukor, Mustafa and Aubakirova, Dana and Russi, Martino and Capuano, Francesco and Pascal, Caroline and Choghari, Jade and Moss, Jess and Wolf, Thomas},
    title = {LeRobot: State-of-the-art Machine Learning for Real-World Robotics in Pytorch},
    howpublished = "\url{https://github.com/huggingface/lerobot}",
    year = {2024}
}
Downloads last month
30
Safetensors
Model size
0.5B params
Tensor type
F32
·
BF16
·
Video Preview
loading

Model tree for Arya945/smolvla_merged

Finetuned
(7659)
this model

Paper for Arya945/smolvla_merged