added DMC Reacher, cartpole, reach_size; removed BBO
This commit is contained in:
@@ -1,4 +1,4 @@
|
||||
from alr_envs.dmc.ball_in_cup.ball_in_the_cup_mp_wrapper import DMCBallInCupMPWrapper
|
||||
from alr_envs.dmc.suite.ball_in_cup.ball_in_cup_mp_wrapper import DMCBallInCupMPWrapper
|
||||
from alr_envs.utils.make_env_helpers import make_dmp_env, make_env
|
||||
|
||||
|
||||
@@ -123,11 +123,11 @@ if __name__ == '__main__':
|
||||
render = False
|
||||
|
||||
# # Standard DMC Suite tasks
|
||||
# example_dmc("fish-swim", seed=10, iterations=1000, render=render)
|
||||
#
|
||||
# # Manipulation tasks
|
||||
# # Disclaimer: The vision versions are currently not integrated and yield an error
|
||||
# example_dmc("manipulation-reach_site_features", seed=10, iterations=250, render=render)
|
||||
example_dmc("fish-swim", seed=10, iterations=1000, render=render)
|
||||
|
||||
# Manipulation tasks
|
||||
# Disclaimer: The vision versions are currently not integrated and yield an error
|
||||
example_dmc("manipulation-reach_site_features", seed=10, iterations=250, render=render)
|
||||
|
||||
# Gym + DMC hybrid task provided in the MP framework
|
||||
example_dmc("dmc_ball_in_cup-catch_detpmp-v0", seed=10, iterations=1, render=render)
|
||||
|
||||
@@ -86,17 +86,18 @@ def example_async(env_id="alr_envs:HoleReacher-v0", n_cpu=4, seed=int('533D', 16
|
||||
|
||||
|
||||
if __name__ == '__main__':
|
||||
render = False
|
||||
# Basic gym task
|
||||
example_general("Pendulum-v0", seed=10, iterations=200, render=True)
|
||||
example_general("Pendulum-v0", seed=10, iterations=200, render=render)
|
||||
#
|
||||
# # Basis task from framework
|
||||
example_general("alr_envs:HoleReacher-v0", seed=10, iterations=200, render=True)
|
||||
example_general("alr_envs:HoleReacher-v0", seed=10, iterations=200, render=render)
|
||||
#
|
||||
# # OpenAI Mujoco task
|
||||
example_general("HalfCheetah-v2", seed=10, render=True)
|
||||
example_general("HalfCheetah-v2", seed=10, render=render)
|
||||
#
|
||||
# # Mujoco task from framework
|
||||
example_general("alr_envs:ALRReacher-v0", seed=10, iterations=200, render=True)
|
||||
example_general("alr_envs:ALRReacher-v0", seed=10, iterations=200, render=render)
|
||||
|
||||
# Vectorized multiprocessing environments
|
||||
example_async(env_id="alr_envs:HoleReacher-v0", n_cpu=2, seed=int('533D', 16), n_samples=2 * 200)
|
||||
|
||||
@@ -148,14 +148,15 @@ def example_fully_custom_mp(seed=1, iterations=1, render=True):
|
||||
|
||||
|
||||
if __name__ == '__main__':
|
||||
render = False
|
||||
# DMP
|
||||
example_mp("alr_envs:HoleReacherDMP-v1", seed=10, iterations=1, render=True)
|
||||
example_mp("alr_envs:HoleReacherDMP-v1", seed=10, iterations=1, render=render)
|
||||
|
||||
# DetProMP
|
||||
example_mp("alr_envs:HoleReacherDetPMP-v1", seed=10, iterations=1, render=True)
|
||||
example_mp("alr_envs:HoleReacherDetPMP-v1", seed=10, iterations=1, render=render)
|
||||
|
||||
# Altered basis functions
|
||||
example_custom_mp("alr_envs:HoleReacherDMP-v1", seed=10, iterations=1, render=True)
|
||||
example_custom_mp("alr_envs:HoleReacherDMP-v1", seed=10, iterations=1, render=render)
|
||||
|
||||
# Custom MP
|
||||
example_fully_custom_mp(seed=10, iterations=1, render=True)
|
||||
example_fully_custom_mp(seed=10, iterations=1, render=render)
|
||||
|
||||
Reference in New Issue
Block a user