SmolVLA Bimanual Pick and Place
Fine-tuned SmolVLA model on BIMAN_PICK_AND_PLACE2 dataset for bimanual manipulation.
Training Details
- Base model: lerobot/smolvla_base
- Dataset: Bcryan/BIMAN_PICK_AND_PLACE2
- Steps: 120,000
- Batch size: 32
Usage
from lerobot.common.policies.smolvla import SmolVLAPolicy
policy = SmolVLAPolicy.from_pretrained("Autobrik/smolvla-biarm-pick-place")
Training Command
python lerobot/scripts/train.py \
--policy.path=lerobot/smolvla_base \
--dataset.repo_id=Bcryan/BIMAN_PICK_AND_PLACE2 \
--batch_size=32 \
--steps=120000
- Downloads last month
- 1