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_norm_fix

Fine-tuned MolmoAct2 (action-expert-only) on SO101 data, with SO101 gripper normalization fix.

Base model allenai/MolmoAct2
Dataset villekuosmanen/armnetbench_block_stack
Task block_stack
Action dim 6 (single-arm)
Cameras top, wrist, front
Normalization MEAN_STD (action + state + gripper), IDENTITY (visual)
Training Isambard GH200, batch 64, bf16, gradient checkpointing
W&B project molmoact2_block_stack
W&B run eqx1ihxu

Checkpoints

Step Path
002000 checkpoints/002000/pretrained_model/

The latest step (002000) is also copied to the repository root for direct loading.

Usage

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

Model tree for pravsels/molmoact2_block_stack_norm_fix

Finetuned
(53)
this model