added DMC Reacher, cartpole, reach_size; removed BBO

This commit is contained in:
ottofabian
2021-07-26 17:07:17 +02:00
parent 7e2f5d664b
commit 24e73d5098
22 changed files with 563 additions and 439 deletions
+6 -6
View File
@@ -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)
+5 -4
View File
@@ -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)