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

Render only one episode

Browse Source
This commit is contained in:
Johannes Fischer
2021-09-13 16:31:24 +02:00
parent 826c0fa219
commit 9b8ceed9c9
1 changed files with 3 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
scratch/etienne/intersimple/gail_options_image.py
View File

@@ -259,8 +259,10 @@ if __name__ == '__main__':
env = NRasterized(**env_settings) env = NRasterized(**env_settings)
for _ in sample(env, generator, None, 'low'): for transition in sample(env, generator, None, 'low'):
env.render() env.render()
if transition['dones']:
break
env.close(filestr='render/'+model_name) env.close(filestr='render/'+model_name)