pravsels commited on
Commit
c22b9f4
·
verified ·
1 Parent(s): 17edd1f

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_fold_tea_towel
12
+
13
+ Fine-tuned [GR00T N1.7-3B](https://huggingface.co/nvidia/GR00T-N1.7-3B) on SO101 data (`fold_tea_towel`).
14
+
15
+ | | |
16
+ |---|---|
17
+ | **Base model** | [nvidia/GR00T-N1.7-3B](https://huggingface.co/nvidia/GR00T-N1.7-3B) |
18
+ | **Dataset** | [villekuosmanen/armnetbench_fold_tea_towel](https://huggingface.co/datasets/villekuosmanen/armnetbench_fold_tea_towel) |
19
+ | **Task** | `fold_tea_towel` |
20
+ | **Training** | Isambard-AI GH200, batch 64, action horizon 16 |
21
+ | **W&B project** | [groot1.7_fold_tea_towel](https://wandb.ai/pravsels/groot1.7_fold_tea_towel) |
22
+ | **W&B run** | [qp91sjl0](https://wandb.ai/pravsels/groot1.7_fold_tea_towel/runs/qp91sjl0) |
23
+
24
+ ## Checkpoints
25
+
26
+ | Step | Loss | Path |
27
+ |------|------|------|
28
+ | 6,000 | 0.0262 | `checkpoints/6000/pretrained_model/` |
29
+ | 4,000 | 0.0432 | `checkpoints/4000/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_fold_tea_towel/checkpoints/6000/pretrained_model"
39
+ ```