From 9b8ceed9c9b5e5c6bcc553b033d0a9742cc14fa0 Mon Sep 17 00:00:00 2001 From: Johannes Fischer Date: Mon, 13 Sep 2021 16:31:24 +0200 Subject: [PATCH] Render only one episode --- scratch/etienne/intersimple/gail_options_image.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/scratch/etienne/intersimple/gail_options_image.py b/scratch/etienne/intersimple/gail_options_image.py index cd91bbe..a4afd58 100644 --- 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)