From 0990eb808ad421a6426c94a9318a5c048e9c2065 Mon Sep 17 00:00:00 2001
From: ebuehrle <43623224+ebuehrle@users.noreply.github.com>
Date: Thu, 4 Nov 2021 11:01:06 +0100
Subject: [PATCH] Larger batches

---
 .../intersimple/gail_options_image_random_location.py       | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

diff --git a/scratch/etienne/intersimple/gail_options_image_random_location.py b/scratch/etienne/intersimple/gail_options_image_random_location.py
index a4a476e..bb033c5 100644
--- a/scratch/etienne/intersimple/gail_options_image_random_location.py
+++ b/scratch/etienne/intersimple/gail_options_image_random_location.py
@@ -25,13 +25,13 @@ 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,
-        expert_batch_size=1024,
+        expert_batch_size=4069,
         discriminator_updates_per_round=10,
         generator_steps=256,
-        generator_total_steps=1024,
+        generator_total_steps=2048,
         generator_updates_per_round=10,
         discount=0.99,
-        epochs=200,
+        epochs=100,
     ):
     env = NRasterizedRouteSpeedRandomAgentLocation(**env_settings)