updated for new mp-pytorch version

This commit is contained in:
Fabian
2023-01-17 09:50:55 +01:00
parent 61626135ff
commit 03b930d7dd
3 changed files with 5 additions and 5 deletions
@@ -33,7 +33,7 @@ def example_mp(env_name="HoleReacherProMP-v0", seed=1, iterations=1, render=True
# Just make sure the correct mode is set before executing the step.
env.render(mode="human")
else:
env.render(mode=None)
env.render()
# Now the action space is not the raw action but the parametrization of the trajectory generator,
# such as a ProMP
@@ -155,7 +155,7 @@ def example_fully_custom_mp(seed=1, iterations=1, render=True):
if __name__ == '__main__':
render = True
render = False
# DMP
example_mp("HoleReacherDMP-v0", seed=10, iterations=5, render=render)
+1 -1
View File
@@ -22,7 +22,7 @@ def example_mp(env_name, seed=1, render=True):
if render and i % 2 == 0:
env.render(mode="human")
else:
env.render(mode=None)
env.render()
ac = env.action_space.sample()
obs, reward, done, info = env.step(ac)
returns += reward