1- # Differential CoDesign
2-
3- Differentiable co-design pipeline for the MUPS hopping robot.
1+ # SurGE: Surrogate Gradient-guided Evolution for Co-design of Legged Robots with Parallel Elasticity
42
53<img src =" docs/diff_codesign.png " width =800/ >
64
7- ## Installation
5+ ## TODO
6+ - [ ] Update readme to a paper-code-release style, i.e. more info about the paper
7+ - [ ] Add more figures from the paper
8+ - [ ] Add bibtex citation
9+ - [ ] Unify naming of the robot, existing names: ` hopper ` , ` mups_robot ` , ` hopper_v2 ` ...
810
9- This repo is self-contained: the locomotion-policy training stack (` legged_gym ` , ` rsl_rl ` ) is
10- vendored under ` src/ ` and installed alongside the codesign package. The only external
11- dependency is [ Isaac Gym] ( https://developer.nvidia.com/isaac-gym ) , which must be installed
12- manually (it is not available on PyPI).
11+
12+ ## Installation
1313
1414``` bash
1515conda env create -f environment.yml
@@ -18,9 +18,6 @@ conda activate codesign
1818pip install -e .
1919```
2020
21- A single ` pip install -e . ` installs three packages — ` mups_codesign ` (the design optimizer),
22- and the vendored ` legged_gym ` and ` rsl_rl ` (policy training). A pretrained policy ships in
23- ` checkpoints/rainbow_v7/ ` , so the codesign scripts run out of the box.
2421
2522## Quick Start
2623
@@ -48,7 +45,7 @@ Plot latest optimization trajectory over last collected objective landscape:
4845``` bash
4946python scripts/plot_landscape.py --policy_id rainbow_v7
5047```
51- <img src =" docs/opt_traj_overlap_landscape.png " width =500/ >
48+ <!-- < img src="docs/opt_traj_overlap_landscape.png" width=500/> -- >
5249
5350Collect gradient vector field of 2D objective landscape from AD:
5451``` bash
@@ -66,7 +63,8 @@ python scripts/plot_gradient_field.py --grad-magnitude 5
6663```
6764Use ` --grad-magnitude ` to scale vector magnitude for minimum overlap.
6865
69- <img src =" docs/gradient_field_ad.png " width =500/ >
66+ <!-- <img src="docs/gradient_field_ad.png" width=500/> -->
67+
7068
7169## Train a Locomotion Policy
7270
0 commit comments