How to use from the
Use from the
LeRobot library
# No code snippets available yet for this library.

# To use this model, check the repository files and the library's documentation.

# Want to help? PRs adding snippets are welcome at:
# https://github.com/huggingface/huggingface.js

molmoact2_block_stack_20k

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

Policy MolmoAct2 (policy.type=molmoact2)
Init checkpoint allenai/MolmoAct2-SO100_101
Dataset villekuosmanen/armnetbench_block_stack
Task block_stack
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_block_stack
W&B project molmoact2_block_stack
W&B run tb8n7j94

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_block_stack_20k")
Downloads last month
13
Safetensors
Model size
5B params
Tensor type
F32
·
BF16
·
Video Preview
loading

Model tree for pravsels/molmoact2_block_stack_20k

Finetuned
(22)
this model