huangfu/InteractionImitation
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Add option to render expert rollout

Browse Source
This commit is contained in:
ebuehrle
2021-09-07 19:19:53 +02:00
parent 88e0b99d7e
commit e9f09cacb7
1 changed files with 14 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

15
scratch/etienne/intersimple/data/expert.py
View File

@@ -74,7 +74,17 @@ class DummyVecEnvPolicy():
states = [p[1] for p in predictions] states = [p[1] for p in predictions]
return actions, states return actions, states
def demonstrations(expert='NormalizedIntersimpleExpert', env='NRasterizedRandomAgent', path=None, min_timesteps=25000, min_episodes=None, env_args={}, policy_args={}): def save_video(env, expert):
env.reset()
env.render()
done = False
while not done:
actions, _ = expert.predict()
_, _, done, _ = env.step(actions)
env.render()
env.close()
def demonstrations(expert='NormalizedIntersimpleExpert', env='NRasterizedRandomAgent', path=None, min_timesteps=25000, min_episodes=None, video=False, env_args={}, policy_args={}):
"""Rollout and save expert demos. """Rollout and save expert demos.
Usage: Usage:
@@ -90,6 +100,9 @@ def demonstrations(expert='NormalizedIntersimpleExpert', env='NRasterizedRandomA
policy = Expert(env, **policy_args) policy = Expert(env, **policy_args)
venv_policy = DummyVecEnvPolicy([lambda: policy]) venv_policy = DummyVecEnvPolicy([lambda: policy])
if video:
save_video(env, policy)
path = path or (policy.__class__.__name__ + '_' + env.__class__.__name__ + '.pkl') path = path or (policy.__class__.__name__ + '_' + env.__class__.__name__ + '.pkl')