Upload folder using huggingface_hub
Browse files- README.md +26 -0
- assets/droid_equad/norm_stats.json +88 -0
- assets/train_config.pkl +3 -0
- metadata.pt +3 -0
- model.safetensors +3 -0
README.md
ADDED
|
@@ -0,0 +1,26 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
---
|
| 2 |
+
license: apache-2.0
|
| 3 |
+
tags:
|
| 4 |
+
- robotics
|
| 5 |
+
- vla
|
| 6 |
+
- pi0
|
| 7 |
+
- droid
|
| 8 |
+
---
|
| 9 |
+
|
| 10 |
+
# MolmoBot-Pi0-DROID-absjoint-reasoninit
|
| 11 |
+
|
| 12 |
+
MolmoBot-Pi0 (PaliGemma-decoder VLA) **full fine-tune** on the equad DROID data
|
| 13 |
+
(`shrg7/droid-equad-stage-1`), with **absolute joint-pose** action targets
|
| 14 |
+
(`arm[t] = q[t+1]`, 7 joints + gripper).
|
| 15 |
+
|
| 16 |
+
**Reasoning-init variant:** before full fine-tuning, a frozen reasoning LoRA
|
| 17 |
+
(trained reasoning-only on the same DROID data) is folded into the PaliGemma
|
| 18 |
+
decoder weights in-place, so the full-FT starts from a reasoning-initialized
|
| 19 |
+
backbone (a weight-space prior), rather than plain base weights.
|
| 20 |
+
|
| 21 |
+
- Config: `molmobot_pi0_lerobot_droid_absjoint`
|
| 22 |
+
- **Checkpoint: step 2000 (intermediate — training in progress)**, held-out
|
| 23 |
+
`val_action_loss = 0.0199`.
|
| 24 |
+
- Contents: `model.safetensors` (flat, full model), `metadata.pt`,
|
| 25 |
+
`assets/droid_equad/norm_stats.json`, `assets/train_config.pkl`.
|
| 26 |
+
Optimizer state omitted.
|
assets/droid_equad/norm_stats.json
ADDED
|
@@ -0,0 +1,88 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"norm_stats": {
|
| 3 |
+
"state": {
|
| 4 |
+
"mean": [
|
| 5 |
+
0.21224477887153625,
|
| 6 |
+
0.17455638945102692,
|
| 7 |
+
0.08217453211545944,
|
| 8 |
+
-2.2301878929138184,
|
| 9 |
+
-0.3177533447742462,
|
| 10 |
+
2.1893584728240967,
|
| 11 |
+
0.5952492952346802,
|
| 12 |
+
0.22651995718479156
|
| 13 |
+
],
|
| 14 |
+
"std": [
|
| 15 |
+
0.1868240088224411,
|
| 16 |
+
0.4287582337856293,
|
| 17 |
+
0.16963618993759155,
|
| 18 |
+
0.3406936824321747,
|
| 19 |
+
0.514870285987854,
|
| 20 |
+
0.398303359746933,
|
| 21 |
+
0.9257502555847168,
|
| 22 |
+
0.33563897013664246
|
| 23 |
+
],
|
| 24 |
+
"q01": [
|
| 25 |
+
-0.10885760933160782,
|
| 26 |
+
-0.6000549793243408,
|
| 27 |
+
-0.20411832630634308,
|
| 28 |
+
-2.6167612075805664,
|
| 29 |
+
-0.9322788119316101,
|
| 30 |
+
1.625175952911377,
|
| 31 |
+
-0.5099077224731445,
|
| 32 |
+
-9.884697227402484e-11
|
| 33 |
+
],
|
| 34 |
+
"q99": [
|
| 35 |
+
0.46970000863075256,
|
| 36 |
+
0.6144798994064331,
|
| 37 |
+
0.33503517508506775,
|
| 38 |
+
-1.8341015577316284,
|
| 39 |
+
0.36206528544425964,
|
| 40 |
+
2.7228288650512695,
|
| 41 |
+
1.7108533382415771,
|
| 42 |
+
0.832818329334259
|
| 43 |
+
]
|
| 44 |
+
},
|
| 45 |
+
"actions": {
|
| 46 |
+
"mean": [
|
| 47 |
+
0.21224477887153625,
|
| 48 |
+
0.17455638945102692,
|
| 49 |
+
0.08217453211545944,
|
| 50 |
+
-2.2301878929138184,
|
| 51 |
+
-0.3177533447742462,
|
| 52 |
+
2.1893584728240967,
|
| 53 |
+
0.5952492952346802,
|
| 54 |
+
0.2692590653896332
|
| 55 |
+
],
|
| 56 |
+
"std": [
|
| 57 |
+
0.1868240088224411,
|
| 58 |
+
0.4287582337856293,
|
| 59 |
+
0.16963618993759155,
|
| 60 |
+
0.3406936824321747,
|
| 61 |
+
0.514870285987854,
|
| 62 |
+
0.398303359746933,
|
| 63 |
+
0.9257502555847168,
|
| 64 |
+
0.3966350555419922
|
| 65 |
+
],
|
| 66 |
+
"q01": [
|
| 67 |
+
-0.10885760933160782,
|
| 68 |
+
-0.6000549793243408,
|
| 69 |
+
-0.20411832630634308,
|
| 70 |
+
-2.6167612075805664,
|
| 71 |
+
-0.9322788119316101,
|
| 72 |
+
1.625175952911377,
|
| 73 |
+
-0.5099077224731445,
|
| 74 |
+
-9.884697227402484e-11
|
| 75 |
+
],
|
| 76 |
+
"q99": [
|
| 77 |
+
0.46970000863075256,
|
| 78 |
+
0.6144798994064331,
|
| 79 |
+
0.33503517508506775,
|
| 80 |
+
-1.8341015577316284,
|
| 81 |
+
0.36206528544425964,
|
| 82 |
+
2.7228288650512695,
|
| 83 |
+
1.7108533382415771,
|
| 84 |
+
0.9533857107162476
|
| 85 |
+
]
|
| 86 |
+
}
|
| 87 |
+
}
|
| 88 |
+
}
|
assets/train_config.pkl
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:025f507e7c8f0d7cb753f9bf1c6eca541b90e6ec7b549fbcac683328f75739a8
|
| 3 |
+
size 1691
|
metadata.pt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:c56cf90a614fdd2e32ea6e352c05b545f484360f11de7e943c3b5e49d8a9b19d
|
| 3 |
+
size 2931
|
model.safetensors
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:60c394e7d6a0218a31460fb7caf68510f46bbe66f9be692cfcbd555d4d17a2ab
|
| 3 |
+
size 7011248944
|