Render only one episode

2021-09-13 16:31:24 +02:00
parent 826c0fa219
commit 9b8ceed9c9
1 changed files with 3 additions and 1 deletions
--- a/scratch/etienne/intersimple/gail_options_image.py
+++ b/scratch/etienne/intersimple/gail_options_image.py
@@ -259,8 +259,10 @@ if __name__ == '__main__':

    env = NRasterized(**env_settings)

-    for _ in sample(env, generator, None, 'low'):
+    for transition in sample(env, generator, None, 'low'):
        env.render()
+        if transition['dones']:
+            break

    env.close(filestr='render/'+model_name)