wassname/ray
1
0
Fork 0
mirror of https://github.com/wassname/ray.git synced 2026-08-15 12:45:23 +08:00
Code Issues Packages Projects Releases Wiki Activity

[RLlib] Dreamer (#10172)

Browse Source
This commit is contained in:
Michael Luo
2020-08-26 13:24:05 +02:00
committed by GitHub
parent 9ca159aa0b
commit 4e9888ce2f
13 changed files with 1362 additions and 69 deletions
Download Patch File Download Diff File Expand all files Collapse all files
rllib/examples/env/dm_control_suite.py
Vendored
+95 -41
View File
@@ -1,85 +1,139 @@
from ray.rllib.env.dm_control_wrapper import DMCEnv
import numpy as np
"""
8 Environments from Deepmind Control Suite
"""
def acrobot_swingup():
def acrobot_swingup(from_pixels=True,
height=64,
width=64,
frame_skip=2,
channels_first=True):
return DMCEnv(
"acrobot",
"swingup",
from_pixels=True,
height=64,
width=64,
task_kwargs={"random": np.random.randint(low=0, high=1e9)})
from_pixels=from_pixels,
height=height,
width=width,
frame_skip=frame_skip,
channels_first=channels_first)
def hopper_hop():
def walker_walk(from_pixels=True,
height=64,
width=64,
frame_skip=2,
channels_first=True):
return DMCEnv(
"walker",
"walk",
from_pixels=from_pixels,
height=height,
width=width,
frame_skip=frame_skip,
channels_first=channels_first)
def hopper_hop(from_pixels=True,
height=64,
width=64,
frame_skip=2,
channels_first=True):
return DMCEnv(
"hopper",
"hop",
from_pixels=True,
height=64,
width=64,
task_kwargs={"random": np.random.randint(low=0, high=1e9)})
from_pixels=from_pixels,
height=height,
width=width,
frame_skip=frame_skip,
channels_first=channels_first)
def hopper_stand():
def hopper_stand(from_pixels=True,
height=64,
width=64,
frame_skip=2,
channels_first=True):
return DMCEnv(
"hopper",
"stand",
from_pixels=True,
height=64,
width=64,
task_kwargs={"random": np.random.randint(low=0, high=1e9)})
from_pixels=from_pixels,
height=height,
width=width,
frame_skip=frame_skip,
channels_first=channels_first)
def cheetah_run():
def cheetah_run(from_pixels=True,
height=64,
width=64,
frame_skip=2,
channels_first=True):
return DMCEnv(
"cheetah",
"run",
from_pixels=True,
height=64,
width=64,
task_kwargs={"random": np.random.randint(low=0, high=1e9)})
from_pixels=from_pixels,
height=height,
width=width,
frame_skip=frame_skip,
channels_first=channels_first)
def walker_run():
def walker_run(from_pixels=True,
height=64,
width=64,
frame_skip=2,
channels_first=True):
return DMCEnv(
"walker",
"run",
from_pixels=True,
height=64,
width=64,
task_kwargs={"random": np.random.randint(low=0, high=1e9)})
from_pixels=from_pixels,
height=height,
width=width,
frame_skip=frame_skip,
channels_first=channels_first)
def pendulum_swingup():
def pendulum_swingup(from_pixels=True,
height=64,
width=64,
frame_skip=2,
channels_first=True):
return DMCEnv(
"pendulum",
"swingup",
from_pixels=True,
height=64,
width=64,
task_kwargs={"random": np.random.randint(low=0, high=1e9)})
from_pixels=from_pixels,
height=height,
width=width,
frame_skip=frame_skip,
channels_first=channels_first)
def cartpole_swingup():
def cartpole_swingup(from_pixels=True,
height=64,
width=64,
frame_skip=2,
channels_first=True):
return DMCEnv(
"cartpole",
"swingup",
from_pixels=True,
height=64,
width=64,
task_kwargs={"random": np.random.randint(low=0, high=1e9)})
from_pixels=from_pixels,
height=height,
width=width,
frame_skip=frame_skip,
channels_first=channels_first)
def humanoid_walk():
def humanoid_walk(from_pixels=True,
height=64,
width=64,
frame_skip=2,
channels_first=True):
return DMCEnv(
"humanoid",
"walk",
from_pixels=True,
height=64,
width=64,
task_kwargs={"random": np.random.randint(low=0, high=1e9)})
from_pixels=from_pixels,
height=height,
width=width,
frame_skip=frame_skip,
channels_first=channels_first)