Upload folder using huggingface_hub
Browse files- .gitattributes +25 -0
- README.md +71 -0
- go2_ac_moe_cts_115k_0.6509.pt +3 -0
- go2_cts_vanilla2_103.5k_0.5786/policy.onnx +3 -0
- go2_cts_vanilla2_103.5k_0.5786/policy.pkl +3 -0
- go2_cts_vanilla2_103.5k_0.5786/policy.pt +3 -0
- go2_dwaq_119.5k_0.5054.pt +3 -0
- go2_him_21k_0.5379.pt +3 -0
- go2_mcp_cts_91k_0.6399/policy.onnx +3 -0
- go2_mcp_cts_91k_0.6399/policy.pt +3 -0
- go2_moe_cts_137000_0.6713/policy.onnx +3 -0
- go2_moe_cts_137000_0.6713/policy.pt +3 -0
- go2_moe_cts_high_slope_thre_164k_0.6715_20260419/model_164000.pt +3 -0
- go2_moe_cts_high_slope_thre_164k_0.6715_20260419/policies/policy.onnx +3 -0
- go2_moe_cts_high_slope_thre_164k_0.6715_20260419/policies/policy.pkl +3 -0
- go2_moe_cts_high_slope_thre_164k_0.6715_20260419/policies/policy.pt +3 -0
- go2_moe_cts_self_103.5k_0.6669/model_103500.pt +3 -0
- go2_moe_cts_self_103.5k_0.6669/policy.onnx +3 -0
- go2_moe_cts_self_103.5k_0.6669/policy.pkl +3 -0
- go2_moe_cts_self_103.5k_0.6669/policy.pt +3 -0
- go2_moe_ng_cts_79k_0.6519/policy.onnx +3 -0
- go2_moe_ng_cts_79k_0.6519/policy.pt +3 -0
.gitattributes
CHANGED
|
@@ -1,5 +1,6 @@
|
|
| 1 |
*.7z filter=lfs diff=lfs merge=lfs -text
|
| 2 |
*.arrow filter=lfs diff=lfs merge=lfs -text
|
|
|
|
| 3 |
*.bin filter=lfs diff=lfs merge=lfs -text
|
| 4 |
*.bz2 filter=lfs diff=lfs merge=lfs -text
|
| 5 |
*.ckpt filter=lfs diff=lfs merge=lfs -text
|
|
@@ -8,6 +9,8 @@
|
|
| 8 |
*.h5 filter=lfs diff=lfs merge=lfs -text
|
| 9 |
*.joblib filter=lfs diff=lfs merge=lfs -text
|
| 10 |
*.lfs.* filter=lfs diff=lfs merge=lfs -text
|
|
|
|
|
|
|
| 11 |
*.mlmodel filter=lfs diff=lfs merge=lfs -text
|
| 12 |
*.model filter=lfs diff=lfs merge=lfs -text
|
| 13 |
*.msgpack filter=lfs diff=lfs merge=lfs -text
|
|
@@ -33,3 +36,25 @@ saved_model/**/* filter=lfs diff=lfs merge=lfs -text
|
|
| 33 |
*.zip filter=lfs diff=lfs merge=lfs -text
|
| 34 |
*.zst filter=lfs diff=lfs merge=lfs -text
|
| 35 |
*tfevents* filter=lfs diff=lfs merge=lfs -text
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
*.7z filter=lfs diff=lfs merge=lfs -text
|
| 2 |
*.arrow filter=lfs diff=lfs merge=lfs -text
|
| 3 |
+
*.avro filter=lfs diff=lfs merge=lfs -text
|
| 4 |
*.bin filter=lfs diff=lfs merge=lfs -text
|
| 5 |
*.bz2 filter=lfs diff=lfs merge=lfs -text
|
| 6 |
*.ckpt filter=lfs diff=lfs merge=lfs -text
|
|
|
|
| 9 |
*.h5 filter=lfs diff=lfs merge=lfs -text
|
| 10 |
*.joblib filter=lfs diff=lfs merge=lfs -text
|
| 11 |
*.lfs.* filter=lfs diff=lfs merge=lfs -text
|
| 12 |
+
*.lz4 filter=lfs diff=lfs merge=lfs -text
|
| 13 |
+
*.mds filter=lfs diff=lfs merge=lfs -text
|
| 14 |
*.mlmodel filter=lfs diff=lfs merge=lfs -text
|
| 15 |
*.model filter=lfs diff=lfs merge=lfs -text
|
| 16 |
*.msgpack filter=lfs diff=lfs merge=lfs -text
|
|
|
|
| 36 |
*.zip filter=lfs diff=lfs merge=lfs -text
|
| 37 |
*.zst filter=lfs diff=lfs merge=lfs -text
|
| 38 |
*tfevents* filter=lfs diff=lfs merge=lfs -text
|
| 39 |
+
# Audio files - uncompressed
|
| 40 |
+
*.pcm filter=lfs diff=lfs merge=lfs -text
|
| 41 |
+
*.sam filter=lfs diff=lfs merge=lfs -text
|
| 42 |
+
*.raw filter=lfs diff=lfs merge=lfs -text
|
| 43 |
+
# Audio files - compressed
|
| 44 |
+
*.aac filter=lfs diff=lfs merge=lfs -text
|
| 45 |
+
*.flac filter=lfs diff=lfs merge=lfs -text
|
| 46 |
+
*.mp3 filter=lfs diff=lfs merge=lfs -text
|
| 47 |
+
*.ogg filter=lfs diff=lfs merge=lfs -text
|
| 48 |
+
*.wav filter=lfs diff=lfs merge=lfs -text
|
| 49 |
+
# Image files - uncompressed
|
| 50 |
+
*.bmp filter=lfs diff=lfs merge=lfs -text
|
| 51 |
+
*.gif filter=lfs diff=lfs merge=lfs -text
|
| 52 |
+
*.png filter=lfs diff=lfs merge=lfs -text
|
| 53 |
+
*.tiff filter=lfs diff=lfs merge=lfs -text
|
| 54 |
+
# Image files - compressed
|
| 55 |
+
*.jpg filter=lfs diff=lfs merge=lfs -text
|
| 56 |
+
*.jpeg filter=lfs diff=lfs merge=lfs -text
|
| 57 |
+
*.webp filter=lfs diff=lfs merge=lfs -text
|
| 58 |
+
# Video files - compressed
|
| 59 |
+
*.mp4 filter=lfs diff=lfs merge=lfs -text
|
| 60 |
+
*.webm filter=lfs diff=lfs merge=lfs -text
|
README.md
ADDED
|
@@ -0,0 +1,71 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
---
|
| 2 |
+
license: mit
|
| 3 |
+
---
|
| 4 |
+
|
| 5 |
+
# Go2 RL GYM Data
|
| 6 |
+
|
| 7 |
+
Checkpoint files for [go2_rl_gym](https://github.com/wty-yy/go2_rl_gym).
|
| 8 |
+
|
| 9 |
+
## Files
|
| 10 |
+
|
| 11 |
+
The table below mirrors the evaluation table in the main repository README.
|
| 12 |
+
`Download` uses relative paths in this dataset repository.
|
| 13 |
+
|
| 14 |
+
| Model | Score | Tracking | Safety | Quality | Level | Download |
|
| 15 |
+
| --- | --- | --- | --- | --- | --- | --- |
|
| 16 |
+
| go2_moe_cts (Ours) | **0.6713** | **0.6669** | **0.7857** | **0.7392** | **7.85** | [go2_moe_cts_137000_0.6713](./go2_moe_cts_137000_0.6713) |
|
| 17 |
+
| go2_ac_moe_cts | 0.6509 | 0.6442 | 0.7644 | 0.7149 | 7.52 | [go2_ac_moe_cts_115k_0.6509.pt](./go2_ac_moe_cts_115k_0.6509.pt) |
|
| 18 |
+
| go2_mcp_cts | 0.6399 | 0.6355 | 0.7542 | 0.7058 | 7.41 | [go2_mcp_cts_91k_0.6399](./go2_mcp_cts_91k_0.6399) |
|
| 19 |
+
| go2_moe_ng_cts | 0.6519 | 0.6447 | 0.7639 | 0.7186 | 7.56 | [go2_moe_ng_cts_79k_0.6519](./go2_moe_ng_cts_79k_0.6519) |
|
| 20 |
+
| [CTS](https://arxiv.org/pdf/2405.10830) vanilla | 0.5786 | 0.5755 | 0.7066 | 0.6624 | 6.83 | [go2_cts_vanilla2_103.5k_0.5786](./go2_cts_vanilla2_103.5k_0.5786) |
|
| 21 |
+
| [HIM](https://github.com/InternRobotics/HIMLoco) | 0.5379 | 0.5453 | 0.6476 | 0.6050 | 6.19 | [go2_him_21k_0.5379.pt](./go2_him_21k_0.5379.pt) |
|
| 22 |
+
| [DreamWaQ](https://arxiv.org/abs/2301.10602) | 0.5054 | 0.5105 | 0.6149 | 0.5730 | 5.74 | [go2_dwaq_119.5k_0.5054.pt](./go2_dwaq_119.5k_0.5054.pt) |
|
| 23 |
+
|
| 24 |
+
Additional self-collision-enabled result:
|
| 25 |
+
|
| 26 |
+
| Model | Score | Download |
|
| 27 |
+
| --- | --- | --- |
|
| 28 |
+
| go2_moe_cts_high_slope_thre | 0.6715 | [go2_moe_cts_high_slope_thre_164k_0.6715_20260419](./go2_moe_cts_high_slope_thre_164k_0.6715_20260419) |
|
| 29 |
+
|
| 30 |
+
## Usage
|
| 31 |
+
|
| 32 |
+
File types:
|
| 33 |
+
|
| 34 |
+
- `policy.pt`: TorchScript policy for Sim2Sim deployment.
|
| 35 |
+
- `policy.onnx`: ONNX policy for C++ / Sim2Real deployment.
|
| 36 |
+
- `policy.pkl`: raw policy weights.
|
| 37 |
+
- `model_*.pt`: full training checkpoint.
|
| 38 |
+
- single `*.pt` files at repository root: standalone checkpoint files kept as released.
|
| 39 |
+
|
| 40 |
+
Example paths:
|
| 41 |
+
|
| 42 |
+
```bash
|
| 43 |
+
# Sim2Sim policy
|
| 44 |
+
./go2_moe_cts_137000_0.6713/policy.pt
|
| 45 |
+
|
| 46 |
+
# ONNX deployment policy
|
| 47 |
+
./go2_moe_cts_137000_0.6713/policy.onnx
|
| 48 |
+
|
| 49 |
+
# Full training checkpoint
|
| 50 |
+
./go2_moe_cts_self_103.5k_0.6669/model_103500.pt
|
| 51 |
+
```
|
| 52 |
+
|
| 53 |
+
For training, evaluation, export, and deployment commands, see the main repository [go2_rl_gym README](https://github.com/wty-yy/go2_rl_gym).
|
| 54 |
+
|
| 55 |
+
## Details
|
| 56 |
+
|
| 57 |
+
- Most directory entries contain exported policies.
|
| 58 |
+
- `go2_moe_cts_high_slope_thre_164k_0.6715_20260419` contains both exported policies and the full `model_164000.pt`.
|
| 59 |
+
- `go2_moe_cts_self_103.5k_0.6669` contains exported policies and `model_103500.pt`.
|
| 60 |
+
|
| 61 |
+
## 📄 Citation
|
| 62 |
+
If you find our work helpful, please cite:
|
| 63 |
+
```bibtex
|
| 64 |
+
@article{wu2026robogauge,
|
| 65 |
+
title={Toward Reliable Sim-to-Real Predictability for MoE-based Robust Quadrupedal Locomotion},
|
| 66 |
+
author={Tianyang Wu and Hanwei Guo and Yuhang Wang and Junshu Yang and Xinyang Sui and Jiayi Xie and Xingyu Chen and Zeyang Liu and Xuguang Lan},
|
| 67 |
+
year={2026},
|
| 68 |
+
journal={arXiv preprint arXiv:2602.00678},
|
| 69 |
+
url={https://arxiv.org/abs/2602.00678},
|
| 70 |
+
}
|
| 71 |
+
```
|
go2_ac_moe_cts_115k_0.6509.pt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:234b419d1eb000f9f4668dc881044adf5824dca53caf370f5aaee8e9f1c11c35
|
| 3 |
+
size 3662890
|
go2_cts_vanilla2_103.5k_0.5786/policy.onnx
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:d1c1592c4b0a786040f8315893e715fb096dd4bbc3a0c571b2054076ecc95dcc
|
| 3 |
+
size 1854636
|
go2_cts_vanilla2_103.5k_0.5786/policy.pkl
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:c31b76e569ce89b7e9bef86523fbaa75b76e32f366ac631484f0dd001aa0aa54
|
| 3 |
+
size 4214814
|
go2_cts_vanilla2_103.5k_0.5786/policy.pt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:3c96d70508aa3a88a633d0a36e36dead30ba8e755e765d23ef19d05dee4a90f2
|
| 3 |
+
size 1864600
|
go2_dwaq_119.5k_0.5054.pt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:e3ad341d73d5cf0222a04aee93973c3065951ce3b14133ca524088a702cdcac9
|
| 3 |
+
size 1067568
|
go2_him_21k_0.5379.pt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:2576a920825e2c2c1a67f3af30281acdd293a2fcc8dea1d654e12de974638096
|
| 3 |
+
size 1067498
|
go2_mcp_cts_91k_0.6399/policy.onnx
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:dd9e91a5d59ea6a026dffc9fc03629a1205d78a65012512144409d065773aaf9
|
| 3 |
+
size 3925692
|
go2_mcp_cts_91k_0.6399/policy.pt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:c5a8a6f1057cc773a44c77b003248a04d5435c9566a6888c0f42ec81dbfd037d
|
| 3 |
+
size 3938346
|
go2_moe_cts_137000_0.6713/policy.onnx
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:bc08253fd68f661bf6be5ce5ca70cec555c3fde2c397d23fbe9814a998e056f5
|
| 3 |
+
size 5191324
|
go2_moe_cts_137000_0.6713/policy.pt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:2206659a6a23886446dcfc05d86888d6cdc9b8a4e31f44cbbc4e2d17ba3d6ead
|
| 3 |
+
size 5206984
|
go2_moe_cts_high_slope_thre_164k_0.6715_20260419/model_164000.pt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:3eb56954fbc48806b243bab559d6a6d5b7253e4254cb9f5cb2070f3f37d39e36
|
| 3 |
+
size 22662338
|
go2_moe_cts_high_slope_thre_164k_0.6715_20260419/policies/policy.onnx
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:522ae37fb15246ebb7a58141e0b0a906ca6d61897810925d43d4fdd0609acb9f
|
| 3 |
+
size 5192804
|
go2_moe_cts_high_slope_thre_164k_0.6715_20260419/policies/policy.pkl
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:a612ceb78f579464eb45d86e0734aa5944ed9ccf73a39913c3f936bfd4df56a6
|
| 3 |
+
size 7550974
|
go2_moe_cts_high_slope_thre_164k_0.6715_20260419/policies/policy.pt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:9d9ad783a1017b6eced5984eb95279cc5b36db8cc84d21e646f46ba2a8023d9d
|
| 3 |
+
size 5208824
|
go2_moe_cts_self_103.5k_0.6669/model_103500.pt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:f4147ba230fd327487dbdd3ec586dff051bc52a5d3d5eddc3c1b688b311575ab
|
| 3 |
+
size 22662338
|
go2_moe_cts_self_103.5k_0.6669/policy.onnx
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:9a52075dfe76604ba6711f0fd487d2699af8487ef97f058869aa38c39a9f1042
|
| 3 |
+
size 5192013
|
go2_moe_cts_self_103.5k_0.6669/policy.pkl
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:f76964534991b2d9a747fcbbe96d8238cd41c487f42b76369b346aa80ade7b2b
|
| 3 |
+
size 7550974
|
go2_moe_cts_self_103.5k_0.6669/policy.pt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:cc949dd2d2dc84704743a86ab0477c3893cbdf0acb9b420a4daa62ef4a96c032
|
| 3 |
+
size 5208760
|
go2_moe_ng_cts_79k_0.6519/policy.onnx
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:ace830456640cb563ad8352ab55f232059d4b8252a12de2c560102089cd00919
|
| 3 |
+
size 5161406
|
go2_moe_ng_cts_79k_0.6519/policy.pt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:32a1cfa189624d73edb1c8573b18edea1643559b814017d5c7b81d4d63d106ca
|
| 3 |
+
size 5177394
|