pravsels commited on
Commit
1ede5a6
·
verified ·
1 Parent(s): 4727e74

Add model card

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