SeonghoonYu commited on
Commit
dbe3ac3
·
verified ·
1 Parent(s): fc3f582

add model card

Browse files
Files changed (1) hide show
  1. README.md +43 -0
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.