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_base_quantile_12k

Fine-tuned MolmoAct2 (action-expert-only) on block_stack with QUANTILES normalization (q01/q99). Intermediate checkpoint at step 12000 while 30k training continues on Isambard.

Policy MolmoAct2 (policy.type=molmoact2)
Init checkpoint allenai/MolmoAct2
Dataset villekuosmanen/armnetbench_block_stack
Task block_stack
Local run molmoact2_block_stack_base_quantile
Checkpoint step 012000 (12k / 30k target)
Normalization QUANTILES (action + state + gripper), IDENTITY (visual)
Training Isambard GH200, batch 64 (16/GPU x 4 DDP), bf16, no gradient checkpointing

Checkpoints

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

Model tree for pravsels/molmoact2_block_stack_base_quantile_12k

Finetuned
(53)
this model