Instructions to use Kaminoikari/smolvla-so101-pickplace-ft with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- LeRobot
How to use Kaminoikari/smolvla-so101-pickplace-ft 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=Kaminoikari/smolvla-so101-pickplace-ft \ --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=Kaminoikari/smolvla-so101-pickplace-ft - Notebooks
- Google Colab
- Kaggle
Upload README.md with huggingface_hub
Browse files
README.md
CHANGED
|
@@ -1,52 +1,52 @@
|
|
| 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 |
-
|
| 42 |
-
|
| 43 |
-
|
| 44 |
-
|
| 45 |
-
|
| 46 |
-
|
| 47 |
-
|
| 48 |
-
|
| 49 |
-
|
| 50 |
-
|
| 51 |
-
|
| 52 |
-
|
|
|
|
| 1 |
+
---
|
| 2 |
+
license: apache-2.0
|
| 3 |
+
library_name: lerobot
|
| 4 |
+
base_model: lerobot/smolvla_base
|
| 5 |
+
datasets:
|
| 6 |
+
- lerobot/svla_so101_pickplace
|
| 7 |
+
tags:
|
| 8 |
+
- robotics
|
| 9 |
+
- lerobot
|
| 10 |
+
- smolvla
|
| 11 |
+
- vision-language-action
|
| 12 |
+
- so-101
|
| 13 |
+
- imitation-learning
|
| 14 |
+
pipeline_tag: robotics
|
| 15 |
+
---
|
| 16 |
+
|
| 17 |
+
# SmolVLA 路 SO-101 Pick-and-Place (fine-tuned)
|
| 18 |
+
|
| 19 |
+
Fine-tune of [`lerobot/smolvla_base`](https://huggingface.co/lerobot/smolvla_base) (450M VLA)
|
| 20 |
+
on the real-robot SO-101 dataset
|
| 21 |
+
[`lerobot/svla_so101_pickplace`](https://huggingface.co/datasets/lerobot/svla_so101_pickplace)
|
| 22 |
+
(50 teleoperated episodes, 2 cameras `up`/`side`, 6-DoF state).
|
| 23 |
+
[`lerobot/svla_so101_pickplace`](https://huggingface.co/datasets/lerobot/svla_so101_pickplace)
|
| 24 |
+
[`lerobot/svla_so101_pickplace`](https://huggingface.co/datasets/lerobot/svla_so101_pickplace)
|
| 25 |
+
[`lerobot/svla_so101_pickplace`](https://huggingface.co/datasets/lerobot/svla_so101_pickplace)
|
| 26 |
+
(50 teleoperated episodes, 2 cameras `up`/`side`, 6-DoF state).
|
| 27 |
+
|
| 28 |
+
## Training
|
| 29 |
+
|
| 30 |
+
| | |
|
| 31 |
+
|---|---|
|
| 32 |
+
| Base | `lerobot/smolvla_base` |
|
| 33 |
+
| Dataset | `lerobot/svla_so101_pickplace` (50 ep / 11,939 frames) |
|
| 34 |
+
| Steps | 2,000 (batch size 8) |
|
| 35 |
+
| GPU | single T4 (~78 min) |
|
| 36 |
+
| Camera mapping | `up to camera1`, `side to camera2` via `--rename_map` |
|
| 37 |
+
| Loss | 0.410 to 0.141 (monotonic) |
|
| 38 |
+
|
| 39 |
+
## Intended use & honest limitations
|
| 40 |
+
|
| 41 |
+
This is a **pipeline-validation / learning run**, not a production policy.
|
| 42 |
+
|
| 43 |
+
- Demonstrates the full real-robot imitation-learning loop: load a real teleoperation dataset, fine-tune a pretrained VLA, converge, ship a checkpoint.
|
| 44 |
+
- Only 2,000 steps (~1.3 epochs). The SmolVLA paper uses ~20k steps; expect this checkpoint to under-perform a fully trained one.
|
| 45 |
+
- No closed-loop success rate. Evaluation on the physical SO-101 arm (`lerobot-record`) was not run (no hardware). Reported signal is training-loss
|
| 46 |
+
convergence only, which proves the model is learning, not real-world task success.
|
| 47 |
+
|
| 48 |
+
## Load
|
| 49 |
+
|
| 50 |
+
```python
|
| 51 |
+
from lerobot.policies.smolvla.modeling_smolvla import SmolVLAPolicy
|
| 52 |
+
policy = SmolVLAPolicy.from_pretrained("Kaminoikari/smolvla-so101-pickplace-ft")
|