File size: 1,368 Bytes
7442164 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 | ---
library_name: lerobot
tags:
- molmoact2
- robotics
- lerobot
- vla
- bimanual
base_model: allenai/MolmoAct2
---
# molmoact2_transfer_cube_norm_fix
Fine-tuned [MolmoAct2](https://huggingface.co/allenai/MolmoAct2) (action-expert-only) on SO101 data, with SO101 gripper normalization fix.
| | |
|---|---|
| **Base model** | [allenai/MolmoAct2](https://huggingface.co/allenai/MolmoAct2) |
| **Dataset** | [villekuosmanen/armnetbench_transfer_cube](https://huggingface.co/datasets/villekuosmanen/armnetbench_transfer_cube) |
| **Task** | `transfer_cube` |
| **Action dim** | 12 (bimanual) |
| **Cameras** | `top`, `left_wrist`, `right_wrist` |
| **Normalization** | MEAN_STD (action + state + gripper), IDENTITY (visual) |
| **Training** | Isambard GH200, batch 64, bf16, gradient checkpointing |
| **W&B project** | [molmoact2_transfer_cube](https://wandb.ai/pravsels/molmoact2_transfer_cube) |
| **W&B run** | [kfileqwv](https://wandb.ai/pravsels/molmoact2_transfer_cube/runs/kfileqwv) |
## Checkpoints
| Step | Path |
|------|------|
| 002000 | `checkpoints/002000/pretrained_model/` |
The latest step (`002000`) is also copied to the **repository root** for direct loading.
## Usage
```python
from lerobot.policies.molmoact2.modeling_molmoact2 import MolmoAct2Policy
policy = MolmoAct2Policy.from_pretrained("pravsels/molmoact2_transfer_cube_norm_fix")
```
|