Skip to content

Commit 35073b6

Browse files
authored
env(gym_control): add gym control classic suite (galilai-group#204)
1 parent 6d5c03b commit 35073b6

22 files changed

Lines changed: 932 additions & 11 deletions

README.md

Lines changed: 16 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -121,11 +121,21 @@ See the full documentation [here](https://galilai-group.github.io/stable-worldmo
121121
</tr>
122122
</table>
123123

124+
<table>
125+
<tr>
126+
<td align="center"><img src="docs/assets/cartpole_control.gif" width="120"/><br><img src="docs/assets/cartpole_control_var.gif" width="120"/></td>
127+
<td align="center"><img src="docs/assets/mountain_car_control.gif" width="120"/><br><img src="docs/assets/mountain_car_control_var.gif" width="120"/></td>
128+
<td align="center"><img src="docs/assets/mountain_car_continuous_control.gif" width="120"/><br><img src="docs/assets/mountain_car_continuous_control_var.gif" width="120"/></td>
129+
<td align="center"><img src="docs/assets/acrobot_control.gif" width="120"/><br><img src="docs/assets/acrobot_control_var.gif" width="120"/></td>
130+
<td align="center"><img src="docs/assets/pendulum_control.gif" width="120"/><br><img src="docs/assets/pendulum_control_var.gif" width="120"/></td>
131+
</tr>
132+
</table>
133+
124134
</div>
125135

126136
<p align="center"><em>Top row: default appearance &nbsp;·&nbsp; Bottom row: visual factor of variation</em></p>
127137

128-
stable-worldmodel supports a large collection of environments from the [DeepMind Control Suite](https://github.com/google-deepmind/dm_control), [OGBench](https://github.com/seohongpark/ogbench), [Craftax](https://github.com/MichaelTMatthews/Craftax), and classical world model benchmarks such as [Two-Room](https://arxiv.org/abs/2411.04983) and [PushT](https://arxiv.org/abs/2303.04137).
138+
stable-worldmodel supports a large collection of environments from the [DeepMind Control Suite](https://github.com/google-deepmind/dm_control), [Gymnasium classic control](https://gymnasium.farama.org/environments/classic_control/), [OGBench](https://github.com/seohongpark/ogbench), [Craftax](https://github.com/MichaelTMatthews/Craftax), and classical world model benchmarks such as [Two-Room](https://arxiv.org/abs/2411.04983) and [PushT](https://arxiv.org/abs/2303.04137).
129139

130140
Each environment includes visual and physical factor variations to evaluate robustness and generalization. New environments can easily be added to stable-worldmodel as they only need to follow the [Gymnasium](https://gymnasium.farama.org/) interface.
131141

@@ -153,6 +163,11 @@ Each environment includes visual and physical factor variations to evaluate robu
153163
| swm/FingerDMControl-v0 | 10 |
154164
| swm/ManipulatorDMControl-v0 | 8 |
155165
| swm/QuadrupedDMControl-v0 | 7 |
166+
| swm/CartPoleControl-v1 | 10 |
167+
| swm/MountainCarControl-v0 | 5 |
168+
| swm/MountainCarContinuousControl-v0 | 4 |
169+
| swm/AcrobotControl-v1 | 11 |
170+
| swm/PendulumControl-v1 | 9 |
156171
| swm/FetchReach-v3 | 8 |
157172
| swm/FetchPush-v3 | 11 |
158173
| swm/FetchSlide-v3 | 11 |

docs/assets/acrobot_control.gif

70.5 KB
Loading
71.3 KB
Loading

docs/assets/cartpole_control.gif

55.3 KB
Loading
52.9 KB
Loading
93.2 KB
Loading
87 KB
Loading
101 KB
Loading
88.8 KB
Loading

docs/assets/pendulum_control.gif

67.6 KB
Loading

0 commit comments

Comments
 (0)