document experiment_cfg + embodiment slot
Browse files
README.md
CHANGED
|
@@ -41,3 +41,28 @@ decent-vla `--resume`, not `from_pretrained`:
|
|
| 41 |
(the LocalRunner end-of-run save, which also records embodiment + backend metadata).
|
| 42 |
Note there is no `server_round_0075.pt`: periodic saves ran every 10 rounds, and 75 is
|
| 43 |
not a multiple of 10, so round 75 exists only as this end-of-run file.
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 41 |
(the LocalRunner end-of-run save, which also records embodiment + backend metadata).
|
| 42 |
Note there is no `server_round_0075.pt`: periodic saves ran every 10 rounds, and 75 is
|
| 43 |
not a multiple of 10, so round 75 exists only as this end-of-run file.
|
| 44 |
+
|
| 45 |
+
## Running these weights (experiment_cfg/)
|
| 46 |
+
|
| 47 |
+
The checkpoint holds ONLY `action_head.*` (537 tensors, 1.6205 B). Three things GR00T
|
| 48 |
+
derives from the training dataset are not in it and not in the public N1.7 base repo, so
|
| 49 |
+
they are published here under `experiment_cfg/`:
|
| 50 |
+
|
| 51 |
+
| file | what it is |
|
| 52 |
+
|---|---|
|
| 53 |
+
| `embodiment_id.json` | the projector-slot mapping used in this run |
|
| 54 |
+
| `processor_config.json` | modality config (state/action index ranges, video keys) + image/state processor settings |
|
| 55 |
+
| `statistics.json` | state/action normalization statistics |
|
| 56 |
+
|
| 57 |
+
**Embodiment slot = 10.** `new_embodiment` is absent from the base repo's
|
| 58 |
+
`embodiment_id.json`; the value comes from `EMBODIMENT_TAG_TO_PROJECTOR_INDEX` in
|
| 59 |
+
`gr00t/model/gr00t_n1d7/processing_gr00t_n1d7.py`. Verified against the weights, not just
|
| 60 |
+
that table: diffing this checkpoint's 14 embodiment-indexed tensors against the base head
|
| 61 |
+
puts **100.0% of the drift in slot 10 and exactly 0.0 in all 31 other slots** (see
|
| 62 |
+
`experiment_cfg/slot_drift.json`). Slot 0 is untrained — falling back to it runs without
|
| 63 |
+
error and produces meaningless actions.
|
| 64 |
+
|
| 65 |
+
Modality (from `processor_config.json`): video keys `front`, `wrist`; state and action
|
| 66 |
+
`single_arm` (0-5) + `gripper` (5-6); action horizon 16; `shortest_image_edge` 256.
|
| 67 |
+
|
| 68 |
+
The training config for this run is included as `experiment_cfg/train_config.yaml`.
|