smolvla_fold_tea_towel

Fine-tuned SmolVLA for bimanual SO101 (12-dim action/state).

Base model lerobot/smolvla_base
Dataset villekuosmanen/armnetbench_fold_tea_towel
Checkpoints 5,000 and 10,000 steps
Training Vast.ai 4xH100, batch 128, ~8.5h total
Action dim 12 (bimanual SO101, 6 per arm)
Cameras top, left_wrist, right_wrist
W&B smolvla_fold_tea_towel

Checkpoints

  • checkpoints/005000/pretrained_model/
  • checkpoints/010000/pretrained_model/

Usage

from lerobot.policies.smolvla.modeling_smolvla import SmolVLAPolicy

policy = SmolVLAPolicy.from_pretrained("pravsels/smolvla_fold_tea_towel", subfolder="checkpoints/010000/pretrained_model")

Training Logs

Full training curves and metrics: https://wandb.ai/pravsels/smolvla_fold_tea_towel/runs/s4u2suf9

Downloads last month
23
Safetensors
Model size
0.5B params
Tensor type
F32
·
BF16
·
Video Preview
loading

Model tree for pravsels/smolvla_fold_tea_towel

Finetuned
(6580)
this model

Dataset used to train pravsels/smolvla_fold_tea_towel