From e37447cafff8e58177d3f79428324041572dcf0f Mon Sep 17 00:00:00 2001 From: ebuehrle <43623224+ebuehrle@users.noreply.github.com> Date: Fri, 29 Oct 2021 16:33:06 +0200 Subject: [PATCH] Increase discriminator batch size and update steps --- .../etienne/intersimple/gail_options_image_random_location.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/scratch/etienne/intersimple/gail_options_image_random_location.py b/scratch/etienne/intersimple/gail_options_image_random_location.py index 3d4a17e..750ca7d 100644 --- a/scratch/etienne/intersimple/gail_options_image_random_location.py +++ b/scratch/etienne/intersimple/gail_options_image_random_location.py @@ -31,10 +31,10 @@ ALL_OPTIONS = [(v,t) for v in [0,2,4,6,8] for t in [5, 10, 20]] # option 0 is sa def train( expert_data, epochs=200, - expert_batch_size=256, + expert_batch_size=1024, generator_steps=1024, discount=0.99, - n_disc_updates_per_round=2, + n_disc_updates_per_round=10, n_gen_updates_per_round=10, ): env = NRasterizedRouteSpeedRandomAgentLocation(**env_settings)