Zhengyiluo commited on
Commit
e9ac930
·
verified ·
1 Parent(s): d598433

Add kinematic planner checkpoint info and download instructions

Browse files
Files changed (1) hide show
  1. README.md +23 -2
README.md CHANGED
@@ -137,9 +137,30 @@ SONIC includes a kinematic planner for real-time locomotion generation — choos
137
 
138
  ## Model Checkpoints
139
 
140
- Pre-trained SONIC policy checkpoints (ONNX format) are available in the `policy/release/` directory of this repository. The models are optimized for real-time inference on NVIDIA Jetson platforms with low-latency control and robust sim-to-real transfer.
141
 
142
- See the [Installation Guide](https://nvlabs.github.io/GR00T-WholeBodyControl/getting_started/installation_deploy.html) for instructions on downloading and using the model checkpoints.
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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