SeonghoonYu commited on
Commit
cd75c28
·
verified ·
1 Parent(s): 8fcd21b

Add H30/H35 loading support (config overlay)

Browse files
h30_h35_support/README.md ADDED
@@ -0,0 +1,19 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ # Loading H30 / H35 RACE2 weights
2
+
3
+ `h30_40k/` and `h35_40k/` are full EMA serving models (step 40000 of `h30_40k_s42` /
4
+ `h35_40k_s42`, action horizon 30 / 35, training seed 42, EMA 0.999) plus the exact
5
+ evaluated slim EMA overlay under `<dir>/overlay/`.
6
+
7
+ Use the OpenPI fork <https://github.com/Seonghoon-Yu/Libero> at revision
8
+ `43bc120d863a2a67fafd3ee2464029ac37040231` with the `h25_support/openpi_overlay/` files
9
+ applied (CPS asset `assets/gauss_timing/vlabench_cps.npz` and the seeded policy-server
10
+ entrypoint are unchanged), and then replace `src/openpi/training/config.py` with the copy in
11
+ this directory's `openpi_overlay/`. The `h25_support` config only registers the H25 configs;
12
+ this one additionally registers `pi05_vlabench_race2_h30`, `pi05_vlabench_race2_h35`
13
+ (and `pi05_vlabench_plainft_h30/h35`). It is a superset of the H25 overlay config.
14
+
15
+ Configs `pi05_vlabench_race2_h30` / `pi05_vlabench_race2_h35` share the H25 recipe, teacher,
16
+ dataset (`SeonghoonYu/vlabench_primitive_ft_lerobot_224`) and CPS labels; only `action_horizon`
17
+ differs. Serve with `--policy.config pi05_vlabench_race2_h30 --policy.dir h30_40k` (replan 30)
18
+ or the H35 equivalents (replan 35). Environment: Python 3.11, PyTorch 2.7.1+cu126,
19
+ Transformers 4.53.2 with the fork's replacement modules, JAX 0.5.3 (`JAX_PLATFORMS=cpu`).
h30_h35_support/openpi_overlay/src/openpi/training/config.py ADDED
The diff for this file is too large to render. See raw diff
 
h30_h35_support/upload_manifest.json ADDED
@@ -0,0 +1,66 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "h30_40k/README.md": {
3
+ "bytes": 1251,
4
+ "sha256": "bef338e7e4277530e4064cf21193b9c8197f6b899759d2b9b7cb9867fb10c804"
5
+ },
6
+ "h30_40k/assets/vlabench/vlabench_ft_primitive/norm_stats.json": {
7
+ "bytes": 4424,
8
+ "sha256": "bce33fc1831830d953ef9bc57e520c02ee44e02b7ad07f27ce92b931a879384f"
9
+ },
10
+ "h30_40k/model.safetensors": {
11
+ "bytes": 7939730180,
12
+ "sha256": "cf9da150ac5c621d4bc263a77c5f3e46dae7c46bfc0712a57739b7d30ad6edd0"
13
+ },
14
+ "h30_40k/overlay/ema.safetensors": {
15
+ "bytes": 2092955412,
16
+ "sha256": "f386ba7b8b7c258c9c565ff187f4c74b78b8ed90d0a5e706cb6d9bced35b5bcb"
17
+ },
18
+ "h30_40k/overlay/metadata.pt": {
19
+ "bytes": 3635,
20
+ "sha256": "1a1682efc584548de15b9fac5f878000cfa6630692de371957e3702917c69522"
21
+ },
22
+ "h30_40k/provenance.json": {
23
+ "bytes": 1647,
24
+ "sha256": "04abac8dee90ec8bc54cf38c83d7f186824e62115d32e67e1a90e29c13741be0"
25
+ },
26
+ "h30_40k/training_config.json": {
27
+ "bytes": 2864,
28
+ "sha256": "915975ed8e58c61c4baa279beb34da83359064370ae4cc94d22262c4b8443d79"
29
+ },
30
+ "h35_40k/README.md": {
31
+ "bytes": 1251,
32
+ "sha256": "6616a53b036bf8d9d98d0af86a3a702d4e779b1898d57a7155b5aade837c8dce"
33
+ },
34
+ "h35_40k/assets/vlabench/vlabench_ft_primitive/norm_stats.json": {
35
+ "bytes": 4424,
36
+ "sha256": "bce33fc1831830d953ef9bc57e520c02ee44e02b7ad07f27ce92b931a879384f"
37
+ },
38
+ "h35_40k/model.safetensors": {
39
+ "bytes": 7939730180,
40
+ "sha256": "6c475bf68a73267f673ac2851a4f9bf9290ba2bab7375135040c5aad8b238236"
41
+ },
42
+ "h35_40k/overlay/ema.safetensors": {
43
+ "bytes": 2092955412,
44
+ "sha256": "8ce31cec60d3ad344785a7f414bc8ce716d07986925278954f7e6e45ac61c3aa"
45
+ },
46
+ "h35_40k/overlay/metadata.pt": {
47
+ "bytes": 3635,
48
+ "sha256": "e28b60bbaa344c118b3875007fa101184ba7c5f539072c6cfa8ef04162ca2574"
49
+ },
50
+ "h35_40k/provenance.json": {
51
+ "bytes": 1647,
52
+ "sha256": "417e781e4dbcf8a76367407f42c523a16d9fca9015b175921da30a2aee25f11c"
53
+ },
54
+ "h35_40k/training_config.json": {
55
+ "bytes": 2865,
56
+ "sha256": "088dd41edb7fc3b781d615788fdb2d9b62e9ba39f801acef688cd4389c329d0e"
57
+ },
58
+ "h30_h35_support/README.md": {
59
+ "bytes": 1315,
60
+ "sha256": "86d90133b27c59a6e61317293d94b9b75606f143aa689cee9adb9deeb11e94d7"
61
+ },
62
+ "h30_h35_support/openpi_overlay/src/openpi/training/config.py": {
63
+ "bytes": 220052,
64
+ "sha256": "a3cc788b0cc2564e0956b9ba32deb17461fb55d59fd0a146b79876d1df236669"
65
+ }
66
+ }