-
Notifications
You must be signed in to change notification settings - Fork 1
/
environments.py
52 lines (40 loc) · 1.08 KB
/
environments.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
# >>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>> MuJoCo environments.
MUJOCO_ROBOTS = [
'InvertedPendulum',
'InvertedDoublePendulum',
'Reacher',
'Hopper',
'HalfCheetah',
'Walker2d',
'Ant',
'Humanoid',
]
MUJOCO_ENVS = ["{}-v2".format(name) for name in MUJOCO_ROBOTS]
MUJOCO_ENVS.extend(["{}-v3".format(name) for name in MUJOCO_ROBOTS])
# >>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>> DM Control environments.
DMC_ROBOTS = [
'Hopper-Hop',
'Cheetah-Run',
'Walker-Walk',
'Walker-Run',
'Stacker-Stack_2',
'Stacker-Stack_4',
'Humanoid-Walk',
'Humanoid-Run',
'Humanoid-Run_Pure_State',
'Humanoid_CMU-Stand',
'Humanoid_CMU-Run',
'Quadruped-Walk',
'Quadruped-Run',
'Quadruped-Escape',
'Quadruped-Fetch',
'Dog-Run',
'Dog-Fetch',
]
DMC_ENVS = ["{}-Feat-v0".format(name) for name in DMC_ROBOTS]
# DMC_ENVS.extend(["{}-Pix-v0".format(name) for name in DMC_ROBOTS]) # XXX: no pixels yet
# >>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>> Aggregate the environments
BENCHMARKS = {
'mujoco': MUJOCO_ENVS,
'dmc': DMC_ENVS,
}