Instructions to use Himanshu77275/smolvla-xarm-hang-blue-mug-v1 with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- LeRobot
How to use Himanshu77275/smolvla-xarm-hang-blue-mug-v1 with LeRobot:
# See https://github.com/huggingface/lerobot?tab=readme-ov-file#installation for more details git clone https://github.com/huggingface/lerobot.git cd lerobot pip install -e .[smolvla]
# Launch finetuning on your dataset python lerobot/scripts/train.py \ --policy.path=Himanshu77275/smolvla-xarm-hang-blue-mug-v1 \ --dataset.repo_id=lerobot/svla_so101_pickplace \ --batch_size=64 \ --steps=20000 \ --output_dir=outputs/train/my_smolvla \ --job_name=my_smolvla_training \ --policy.device=cuda \ --wandb.enable=true
# Run the policy using the record function python -m lerobot.record \ --robot.type=so101_follower \ --robot.port=/dev/ttyACM0 \ # <- Use your port --robot.id=my_blue_follower_arm \ # <- Use your robot id --robot.cameras="{ front: {type: opencv, index_or_path: 8, width: 640, height: 480, fps: 30}}" \ # <- Use your cameras --dataset.single_task="Grasp a lego block and put it in the bin." \ # <- Use the same task description you used in your dataset recording --dataset.repo_id=HF_USER/dataset_name \ # <- This will be the dataset name on HF Hub --dataset.episode_time_s=50 \ --dataset.num_episodes=10 \ --policy.path=Himanshu77275/smolvla-xarm-hang-blue-mug-v1 - Notebooks
- Google Colab
- Kaggle
SmolVLA xArm Hang Blue Mug v1
This repository contains a SmolVLA policy fine-tuned with LeRobot on the JayCao99/xarm-hang-blue-mug-v1 dataset.
Training Summary
| Field | Value |
|---|---|
| Base policy | lerobot/smolvla_base |
| Dataset | JayCao99/xarm-hang-blue-mug-v1 |
| Robot | xArm7 |
| Episodes | 200 |
| Frames | 132,214 |
| Cameras | ego, wrist |
| State dim | 15 |
| Action dim | 8 |
| Training steps | 100,000 |
| Batch size | 32 |
| Save frequency | 5,000 steps |
| GPU | 1x A100 80GB, Unity superpod-a100 |
| Runtime | approximately 7h 28m |
| Peak VRAM from log | approximately 7.69 GB |
| WandB | disabled |
Checkpoints
| Checkpoint | Notes |
|---|---|
checkpoint-100000/ |
Final policy checkpoint after 100k finetuning steps |
Usage
from huggingface_hub import snapshot_download
ckpt_dir = snapshot_download(
repo_id="Himanshu77275/smolvla-xarm-hang-blue-mug-v1",
allow_patterns="checkpoint-100000/*",
)
The checkpoint directory contains the LeRobot policy config, model weights, and dataset normalization/preprocessing files needed for policy loading.