This repository contains the final checkpoint for a SmolVLA policy fine-tune trained on aswinkumar99/task2-known for SO101 sponge pick-and-place experiments.

Dataset meaning: Task 2: Multiple Sponges - No Distractors (fixed-locations / known).

This SmolVLA policy is a fine-tune of lerobot/smolvla_base, as recorded by both the launch command (--policy.path=lerobot/smolvla_base) and the saved training config (pretrained_path: lerobot/smolvla_base).

Training Setup

Dataset repo: aswinkumar99/task2-known
Local dataset root during training: /home/riftuser/datasets_combined/aswinkumar99/task2-known
Output directory during training: /home/riftuser/outputs_matrix/smolvla/task2-known_bs64_s20000
Batch size: 64
Training steps: 20000
Checkpoint save frequency: 5000
Data loader workers: 8
WandB project: so101-layout-generalization
GPU: NVIDIA RTX PRO 6000 Blackwell Max-Q Workstation Edition
Python: CPython 3.12.13
CUDA: 12.9
Training start: 2026-04-24T00:24:52.956573+00:00
Training end: 2026-04-24T02:49:47
Approximate training duration: 2h 24m 54s
Base model: lerobot/smolvla_base
Observation camera rename map: {"observation.images.overhead": "observation.images.camera1", "observation.images.wrist": "observation.images.camera2"}
Action chunk size: 50
Action steps predicted: 50

Exact Training Command

bash

lerobot-train \
  --dataset.repo_id=aswinkumar99/task2-known \
  --dataset.root=/home/riftuser/datasets_combined/aswinkumar99/task2-known \
  --dataset.video_backend=torchcodec \
  --output_dir=/home/riftuser/outputs_matrix/smolvla/task2-known_bs64_s20000 \
  --job_name=smolvla_task2-known_bs64 \
  --batch_size=64 \
  --steps=20000 \
  --log_freq=200 \
  --save_freq=5000 \
  --save_checkpoint=true \
  --num_workers=8 \
  --wandb.enable=true \
  --wandb.project=so101-layout-generalization \
  --wandb.mode=online \
  --wandb.disable_artifact=true \
  --policy.path=lerobot/smolvla_base \
  --policy.device=cuda \
  --policy.push_to_hub=false \
  --rename_map={"observation.images.overhead": "observation.images.camera1", "observation.images.wrist": "observation.images.camera2"}

Repository Contents

pretrained_model/: final downloadable model artifacts for inference/loading
training_state/: optimizer, RNG, scheduler/state, and step information for resuming or auditability

Notes

This repo stores the final checkpoint that was uploaded from the cloud training workspace.
The checkpoint was trained with LeRobot tooling via lerobot-train.
For SO101 experiments in this workspace, the dataset source was created by Aswinkumar.

Creator

Aswinkumar

Website: aswinkumar.me
Hugging Face repo: https://huggingface.co/aswinkumar99/LeRobot-SO101-SmolVLA-task2-known_bs64_s20000

aswinkumar99/LeRobot-SO101-SmolVLA-task2-known_bs64_s20000

LeRobot SO101 SmolVLA task2-known_bs64_s20000

Summary

Training Setup

Exact Training Command

Repository Contents

Notes

Creator

aswinkumar99/
LeRobot-SO101-SmolVLA-task2-known_bs64_s20000