Downloads · 30 days
11
17% of all-time downloads
taewonkoo/VLA-Adapter-SO101-push-cube-step-10000
VLA-Adapter-SO101-push-cube-step-10000 is a robotics model from taewonkoo. Use it for the robotics task on the model card, and read the license before you ship it in a product. It is set up for transformers.
This repository contains an intermediate 10,000-optimizer-step checkpoint of VLA-Adapter fine-tuned for the SO-101 pushcube task.
Downloads · 30 days
11
17% of all-time downloads
All-time downloads
64
Public
Parameters
1.3B
4.6 GB on disk
Likes
0
Public
Click a slice to open those files.
.safetensors3 GB · 65%
From the Hugging Face model README
This repository contains an intermediate 10,000-optimizer-step checkpoint of
VLA-Adapter fine-tuned for the SO-101 push_cube task.
| Setting | Value |
|---|---|
| Dataset | taewonkoo/push_cube_v2_rlds |
| Optimizer step | 10,000 / 20,000 |
| Per-device batch size | 4 |
| Gradient accumulation | 4 |
| Effective batch size | 16 |
| Learning rate | 1e-4 |
| Scheduler | Cosine with 2,000 warmup steps |
| LoRA rank | 32 |
| Input images | 2 (front, top) |
| Proprioception | Enabled, 6 dimensions |
| Action dimension | 6 |
| Action chunk length | 8 |
| Precision | BF16 on an RTX 3090 |
The RLDS dataset contains 50 trajectories and 13,679 transitions. Actions and proprioceptive states follow the six-motor SO-101 representation.
model.safetensors: VLA model with the LoRA weights mergedlora_adapter/: trainable LoRA adapteraction_head--10000_checkpoint.pt: continuous action headproprio_projector--10000_checkpoint.pt: proprioception projectorvla_extras--10000_checkpoint.pt: trainable VLA extras such as action queriestraining_state--10000_checkpoint.pt: optimizer, scheduler, scaler, and RNG stateThe separate action head and proprio projector are required for the SO-101
continuous-control policy. The merged model.safetensors alone is not the
complete policy.
Use the SO-101-compatible training and inference code from:
https://github.com/TaewonKoo/VLA-Adapter-SO101
This is an intermediate training checkpoint, not the final 20,000-step model. Review the base model, dataset, and code licenses before redistribution or deployment. Validate the policy in simulation or a controlled robot workspace before operating physical hardware.