pravsels commited on
Commit
faa5cd3
·
verified ·
1 Parent(s): 23de329

Update model card

Browse files
Files changed (1) hide show
  1. README.md +38 -0
README.md ADDED
@@ -0,0 +1,38 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ library_name: lerobot
3
+ tags:
4
+ - molmoact2
5
+ - robotics
6
+ - lerobot
7
+ - vla
8
+
9
+ base_model: allenai/MolmoAct2-SO100_101
10
+ ---
11
+
12
+ # molmoact2_tool_insert_20k
13
+
14
+ Fine-tuned [MolmoAct2](https://huggingface.co/allenai/MolmoAct2) (action-expert-only) for `tool_insert` on SO101 data.
15
+
16
+ | | |
17
+ |---|---|
18
+ | **Policy** | MolmoAct2 (`policy.type=molmoact2`) |
19
+ | **Init checkpoint** | [allenai/MolmoAct2-SO100_101](https://huggingface.co/allenai/MolmoAct2-SO100_101) |
20
+ | **Dataset** | [villekuosmanen/armnetbench_tool_insert](https://huggingface.co/datasets/villekuosmanen/armnetbench_tool_insert) |
21
+ | **Task** | `tool_insert` |
22
+ | **Action dim** | 6 (single-arm) |
23
+ | **Cameras** | `top`, `wrist`, `front` |
24
+ | **Training** | 20k steps, batch 32 (8/GPU × 4 DDP), 4× RTX 5090 on Vast.ai |
25
+ | **Prior HF repo** | [pravsels/molmoact2_tool_insert](https://huggingface.co/pravsels/molmoact2_tool_insert) |
26
+ | **W&B project** | [molmoact2_tool_insert](https://wandb.ai/pravsels/molmoact2_tool_insert) |
27
+ | **W&B run** | [vsh6rneo](https://wandb.ai/pravsels/molmoact2_tool_insert/runs/vsh6rneo) |
28
+
29
+ ## Checkpoints
30
+
31
+ The final checkpoint (local step `020000`, 20k total steps) lives at the **repository root** for direct loading.
32
+
33
+ ## Usage
34
+
35
+ ```python
36
+ from lerobot.policies.molmoact2.modeling_molmoact2 import MolmoAct2Policy
37
+ policy = MolmoAct2Policy.from_pretrained("pravsels/molmoact2_tool_insert_20k")
38
+ ```