Update model card: rename model files, fix download script flags
Browse files
README.md
CHANGED
|
@@ -141,8 +141,8 @@ All checkpoints (ONNX format) are available directly in this repository. Inferen
|
|
| 141 |
|
| 142 |
| Checkpoint | File | Description |
|
| 143 |
|---|---|---|
|
| 144 |
-
| Policy encoder | `
|
| 145 |
-
| Policy decoder | `
|
| 146 |
| Kinematic planner | `planner_sonic.onnx` | Real-time locomotion style planner |
|
| 147 |
|
| 148 |
**Quick download** (requires `pip install huggingface_hub`):
|
|
@@ -155,8 +155,8 @@ snapshot_download(repo_id="nvidia/GEAR-SONIC", local_dir="gear_sonic_deploy")
|
|
| 155 |
Or use the download script from the GitHub repo:
|
| 156 |
|
| 157 |
```bash
|
| 158 |
-
python download_from_hf.py
|
| 159 |
-
python download_from_hf.py --planner # policy
|
| 160 |
```
|
| 161 |
|
| 162 |
See the [Download Models guide](https://nvlabs.github.io/GR00T-WholeBodyControl/getting_started/download_models.html) for full instructions.
|
|
|
|
| 141 |
|
| 142 |
| Checkpoint | File | Description |
|
| 143 |
|---|---|---|
|
| 144 |
+
| Policy encoder | `model_encoder.onnx` | Encodes motion reference into latent |
|
| 145 |
+
| Policy decoder | `model_decoder.onnx` | Decodes latent into joint actions |
|
| 146 |
| Kinematic planner | `planner_sonic.onnx` | Real-time locomotion style planner |
|
| 147 |
|
| 148 |
**Quick download** (requires `pip install huggingface_hub`):
|
|
|
|
| 155 |
Or use the download script from the GitHub repo:
|
| 156 |
|
| 157 |
```bash
|
| 158 |
+
python download_from_hf.py # policy + planner (default)
|
| 159 |
+
python download_from_hf.py --no-planner # policy only
|
| 160 |
```
|
| 161 |
|
| 162 |
See the [Download Models guide](https://nvlabs.github.io/GR00T-WholeBodyControl/getting_started/download_models.html) for full instructions.
|