molmoact2_cable_unclip_quantile_norm_fix_25k

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

Policy MolmoAct2 (policy.type=molmoact2)
Init checkpoint allenai/MolmoAct2
Dataset pravsels/cable_unclip_remote
Task cable_unclip
Action dim 6 (single-arm)
Cameras top, wrist, front
Training 25k steps, QUANTILES norm, freeze, batch 32 global, Isambard GH200
Prior HF repo pravsels/molmoact2_cable_unclip
W&B project molmoact2_cable_unclip_quantile_norm_fix_25k
W&B run 79ymd2hy

Checkpoints

The checkpoint (local step 025000, 25k training steps) lives at the repository root for direct loading.

Verification

| Checkpoint step | 025000 | | Source path | checkpoints/025000/pretrained_model/ | | model.safetensors | 10,884,573,720 bytes, sha256 aa63dd389c815da7a35ae12fec466f47b8fdad60f7a1c948728c0caaf3253ed7 | | policy_preprocessor.json | 2,495 bytes, sha256 e7c8b8293cb0265a01f83278033272efcb21b4c7bfb031cfbd683ed74ee7b139 | | policy_postprocessor.json | 757 bytes, sha256 6dbed1e1ec69e8c50f3a04c1f144a54231e3ef508f15fd7896ead43ea645b033 | | train_config.json | 8,250 bytes, sha256 39a5da07eb1f6cc2d8c741452eef363bab03c7a193d75333c01d669cce2317d0 |

Verify after download:

sha256sum model.safetensors
# expected: aa63dd389c815da7a35ae12fec466f47b8fdad60f7a1c948728c0caaf3253ed7

Usage

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

Model tree for pravsels/molmoact2_cable_unclip_quantile_norm_fix_25k

Finetuned
(37)
this model