molmoact2_cable_unclip_20k

Fine-tuned MolmoAct2 (action-expert-only) for cable_unclip on SO101 data.

Policy MolmoAct2 (policy.type=molmoact2)
Init checkpoint allenai/MolmoAct2-SO100_101
Dataset pravsels/cable_unclip_remote
Task cable_unclip
Action dim 6 (single-arm)
Cameras top, wrist, front
Training 20k steps, batch 32 (8/GPU × 4 DDP), 4× RTX 5090 on Vast.ai
Prior HF repo pravsels/molmoact2_cable_unclip
W&B project molmoact2_cable_unclip
W&B run brwy71w8

Checkpoints

The final checkpoint (local step 020000, 20k total steps) lives at the repository root for direct loading.

Usage

from lerobot.policies.molmoact2.modeling_molmoact2 import MolmoAct2Policy
policy = MolmoAct2Policy.from_pretrained("pravsels/molmoact2_cable_unclip_20k")
Downloads last month
29
Safetensors
Model size
5B params
Tensor type
F32
·
BF16
·
Video Preview
loading

Model tree for pravsels/molmoact2_cable_unclip_20k

Finetuned
(14)
this model