Commit ·
9a723b7
1
Parent(s): 4c94677
Update README with deploy.yaml instructions and build steps
Browse filesCo-Authored-By: Claude Opus 4.6 (1M context) <noreply@anthropic.com>
README.md
CHANGED
|
@@ -43,6 +43,7 @@ RL locomotion policy for the [Unitree Go2](https://www.unitree.com/go2/) quadrup
|
|
| 43 |
| `policy.onnx` + `policy.onnx.data` | ONNX model for deployment (go2_ctrl) |
|
| 44 |
| `model_500.pt` | Final PyTorch checkpoint (best for fine-tuning) |
|
| 45 |
| `model_0.pt` ... `model_400.pt` | Intermediate checkpoints every 100 steps |
|
|
|
|
| 46 |
| `params/env.yaml` | Environment configuration |
|
| 47 |
| `params/agent.yaml` | Agent/PPO configuration |
|
| 48 |
| `events.out.tfevents.*` | TensorBoard training logs |
|
|
@@ -52,15 +53,24 @@ RL locomotion policy for the [Unitree Go2](https://www.unitree.com/go2/) quadrup
|
|
| 52 |
### Deploy in MuJoCo simulator
|
| 53 |
|
| 54 |
```bash
|
| 55 |
-
# Copy ONNX model
|
| 56 |
cp policy.onnx policy.onnx.data \
|
| 57 |
unitree_rl_mjlab/deploy/robots/go2/config/policy/velocity/v0/exported/
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 58 |
|
| 59 |
# Run simulator + controller
|
| 60 |
cd unitree_mujoco/simulate/build && ./unitree_mujoco
|
| 61 |
cd unitree_rl_mjlab/deploy/robots/go2/build && ./go2_ctrl --network=lo
|
| 62 |
```
|
| 63 |
|
|
|
|
|
|
|
| 64 |
### Fine-tune on rough terrain
|
| 65 |
|
| 66 |
```bash
|
|
|
|
| 43 |
| `policy.onnx` + `policy.onnx.data` | ONNX model for deployment (go2_ctrl) |
|
| 44 |
| `model_500.pt` | Final PyTorch checkpoint (best for fine-tuning) |
|
| 45 |
| `model_0.pt` ... `model_400.pt` | Intermediate checkpoints every 100 steps |
|
| 46 |
+
| `params/deploy.yaml` | Deploy configuration (obs order, action scale, joint mapping) |
|
| 47 |
| `params/env.yaml` | Environment configuration |
|
| 48 |
| `params/agent.yaml` | Agent/PPO configuration |
|
| 49 |
| `events.out.tfevents.*` | TensorBoard training logs |
|
|
|
|
| 53 |
### Deploy in MuJoCo simulator
|
| 54 |
|
| 55 |
```bash
|
| 56 |
+
# Copy ONNX model + deploy config
|
| 57 |
cp policy.onnx policy.onnx.data \
|
| 58 |
unitree_rl_mjlab/deploy/robots/go2/config/policy/velocity/v0/exported/
|
| 59 |
+
cp params/deploy.yaml \
|
| 60 |
+
unitree_rl_mjlab/deploy/robots/go2/config/policy/velocity/v0/params/
|
| 61 |
+
|
| 62 |
+
# Build controller
|
| 63 |
+
cd unitree_rl_mjlab/deploy/robots/go2
|
| 64 |
+
mkdir -p build && cd build
|
| 65 |
+
cmake .. -DCMAKE_BUILD_TYPE=Release && make -j$(nproc)
|
| 66 |
|
| 67 |
# Run simulator + controller
|
| 68 |
cd unitree_mujoco/simulate/build && ./unitree_mujoco
|
| 69 |
cd unitree_rl_mjlab/deploy/robots/go2/build && ./go2_ctrl --network=lo
|
| 70 |
```
|
| 71 |
|
| 72 |
+
> **Important:** This model was trained **without `gait_phase`** and with **action scale 0.5**. The default `deploy.yaml` in unitree_rl_mjlab may differ — use `params/deploy.yaml` from this repo.
|
| 73 |
+
|
| 74 |
### Fine-tune on rough terrain
|
| 75 |
|
| 76 |
```bash
|