pravsels's picture
Add model card
c6547a3 verified
|
Raw
History Blame Contribute Delete
1.09 kB
metadata
library_name: lerobot
tags:
  - molmoact2
  - robotics
  - lerobot
  - vla
base_model: allenai/MolmoAct2

molmoact2_cable_unclip

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

Base model allenai/MolmoAct2
Dataset pravsels/cable_unclip_remote
Task cable_unclip
Training Vast.ai H100 SXM, batch 64, bf16, gradient checkpointing
W&B project molmoact2_cable_unclip
W&B run is0gukiv

Checkpoints

Step Path
006000 checkpoints/006000/pretrained_model/
004000 checkpoints/004000/pretrained_model/

Usage

from lerobot.policies.molmoact2.modeling_molmoact2 import MolmoAct2Policy
policy = MolmoAct2Policy.from_pretrained("pravsels/molmoact2_cable_unclip", revision="main")