Add kinematic planner checkpoint info and download instructions
Browse files
README.md
CHANGED
|
@@ -137,9 +137,30 @@ SONIC includes a kinematic planner for real-time locomotion generation — choos
|
|
| 137 |
|
| 138 |
## Model Checkpoints
|
| 139 |
|
| 140 |
-
|
| 141 |
|
| 142 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 143 |
|
| 144 |
## Documentation
|
| 145 |
|
|
|
|
| 137 |
|
| 138 |
## Model Checkpoints
|
| 139 |
|
| 140 |
+
All checkpoints (ONNX format) are available directly in this repository and optimized for real-time inference on NVIDIA Jetson platforms.
|
| 141 |
|
| 142 |
+
| Checkpoint | Path in repo | Description |
|
| 143 |
+
|---|---|---|
|
| 144 |
+
| Policy encoder | `policy/release/model_releasev1-1_encoder.onnx` | Encodes motion reference into latent |
|
| 145 |
+
| Policy decoder | `policy/release/model_releasev1-1_decoder.onnx` | Decodes latent into joint actions |
|
| 146 |
+
| Observation config | `policy/release/observation_config.yaml` | Sensor / obs configuration |
|
| 147 |
+
| Kinematic planner | `planner/target_vel/V2/planner_sonic.onnx` | Real-time locomotion style planner |
|
| 148 |
+
|
| 149 |
+
**Quick download** (requires `pip install huggingface_hub`):
|
| 150 |
+
|
| 151 |
+
```python
|
| 152 |
+
from huggingface_hub import snapshot_download
|
| 153 |
+
snapshot_download(repo_id="nvidia/GEAR-SONIC", local_dir="gear_sonic_deploy")
|
| 154 |
+
```
|
| 155 |
+
|
| 156 |
+
Or use the download script from the GitHub repo:
|
| 157 |
+
|
| 158 |
+
```bash
|
| 159 |
+
python download_from_hf.py # policy only
|
| 160 |
+
python download_from_hf.py --planner # policy + planner
|
| 161 |
+
```
|
| 162 |
+
|
| 163 |
+
See the [Download Models guide](https://nvlabs.github.io/GR00T-WholeBodyControl/getting_started/download_models.html) for full instructions.
|
| 164 |
|
| 165 |
## Documentation
|
| 166 |
|