pravsels commited on
Commit
3903a9a
·
verified ·
1 Parent(s): 0ee9a5e

Add model card with W&B link

Browse files
Files changed (1) hide show
  1. README.md +39 -0
README.md ADDED
@@ -0,0 +1,39 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ library_name: gr00t
3
+ tags:
4
+ - gr00t
5
+ - gr00t-n1.7
6
+ - robotics
7
+ - vla
8
+ base_model: nvidia/GR00T-N1.7-3B
9
+ ---
10
+
11
+ # groot1.7_block_stack
12
+
13
+ Fine-tuned [GR00T N1.7-3B](https://huggingface.co/nvidia/GR00T-N1.7-3B) on SO101 data (`block_stack`).
14
+
15
+ | | |
16
+ |---|---|
17
+ | **Base model** | [nvidia/GR00T-N1.7-3B](https://huggingface.co/nvidia/GR00T-N1.7-3B) |
18
+ | **Dataset** | [villekuosmanen/armnetbench_block_stack](https://huggingface.co/datasets/villekuosmanen/armnetbench_block_stack) |
19
+ | **Task** | `block_stack` |
20
+ | **Training** | Isambard-AI GH200, batch 64, action horizon 16 |
21
+ | **W&B project** | [groot1.7_block_stack](https://wandb.ai/pravsels/groot1.7_block_stack) |
22
+ | **W&B run** | [ypeedac1](https://wandb.ai/pravsels/groot1.7_block_stack/runs/ypeedac1) |
23
+
24
+ ## Checkpoints
25
+
26
+ | Step | Loss | Path |
27
+ |------|------|------|
28
+ | 10,000 | 0.0194 | `checkpoints/10000/pretrained_model/` |
29
+ | 8,000 | 0.0262 | `checkpoints/8000/pretrained_model/` |
30
+
31
+ ## Usage
32
+
33
+ Load from a specific step:
34
+
35
+ ```python
36
+ from pathlib import Path
37
+ # Use checkpoints/<step>/pretrained_model/ with the GR00T inference stack
38
+ ckpt = "pravsels/groot1.7_block_stack/checkpoints/10000/pretrained_model"
39
+ ```