add model card
Browse files
README.md
ADDED
|
@@ -0,0 +1,43 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
---
|
| 2 |
+
license: apache-2.0
|
| 3 |
+
tags:
|
| 4 |
+
- robotics
|
| 5 |
+
- robotwin
|
| 6 |
+
- pi0.5
|
| 7 |
+
- vla
|
| 8 |
+
---
|
| 9 |
+
|
| 10 |
+
# RACE RoboTwin — π0.5 t5k showcase checkpoints
|
| 11 |
+
|
| 12 |
+
Single-task π0.5 (OpenPI) checkpoints on RoboTwin 2.0 (aloha-agilex, `demo_clean`, 50 demos/task), for 6 tasks:
|
| 13 |
+
`pick_dual_bottles`, `move_can_pot`, `place_dual_shoes`, `place_can_basket`, `blocks_ranking_rgb`, `stack_blocks_three`.
|
| 14 |
+
|
| 15 |
+
All fine-tunes start from the **5k-step teacher** (weak-teacher regime, "t5k").
|
| 16 |
+
|
| 17 |
+
## Layout
|
| 18 |
+
|
| 19 |
+
```
|
| 20 |
+
<task>/
|
| 21 |
+
teacher_ac50/ # 5k-step teacher (JAX -> PyTorch conversion, float32)
|
| 22 |
+
vlmfreeze_ac50/{5000,10000}/ # phase-loc fine-tune, VLM frozen, action horizon 50
|
| 23 |
+
vlmfreeze_ac75/{5000,10000}/ # action horizon 75
|
| 24 |
+
vlmfreeze_ac100/{5000,10000}/ # action horizon 100
|
| 25 |
+
```
|
| 26 |
+
|
| 27 |
+
Each checkpoint dir has `model.safetensors`, `assets/` (norm stats), and `metadata.pt`
|
| 28 |
+
(training config + step). Optimizer states are omitted.
|
| 29 |
+
|
| 30 |
+
## Serving / eval
|
| 31 |
+
|
| 32 |
+
Serve with OpenPI's `scripts/serve_policy.py` (PyTorch path):
|
| 33 |
+
|
| 34 |
+
```
|
| 35 |
+
python scripts/serve_policy.py --port <P> policy:checkpoint \
|
| 36 |
+
--policy.config=st_<task>_phase_loc[_h75|_h100] --policy.dir=<downloaded dir>
|
| 37 |
+
```
|
| 38 |
+
|
| 39 |
+
`--policy.config` must match the action horizon (`st_<t>_phase_loc` = 50,
|
| 40 |
+
`_h75` = 75, `_h100` = 100); evaluate with RoboTwin 2.0 `script/eval_policy.py`
|
| 41 |
+
(`--pi0_step` = the same horizon, `demo_clean`, seed 0, 50 episodes).
|
| 42 |
+
|
| 43 |
+
Checkpoints are uploaded progressively as trainings/evals complete.
|