From 4c8fb77a919e8990dd84f7997cdac902415d50ae Mon Sep 17 00:00:00 2001
From: ebuehrle <43623224+ebuehrle@users.noreply.github.com>
Date: Sat, 6 Nov 2021 18:49:00 +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 0e23f5f..63ad89d 100644
--- a/scratch/etienne/intersimple/gail_options_image_random_location.py
+++ b/scratch/etienne/intersimple/gail_options_image_random_location.py
@@ -46,11 +46,11 @@ class LLBuffer(deque):
 
 def train(
         expert_data,
-        expert_batch_size=3072,
+        expert_batch_size=4096,
         discriminator_updates_per_round=20,
         generator_steps=1024,
         generator_batch_size=1024,
-        generator_total_steps=4096,
+        generator_total_steps=8192,
         generator_updates_per_round=10,
         discount=1.0,
         epochs=200,
@@ -138,7 +138,7 @@ def evaluate():
 
 # %%
 if __name__ == '__main__':
-    with open("data/NormalizedIntersimpleExpertMu.001N100000_TLNRasterizedRouteRandomAgentLocationw70h70mppx1mu.001rskips50.pkl", "rb") as f:
+    with open("data/NormalizedIntersimpleExpertMu.001N50000_TLNRasterizedRouteRandomAgentLocationw70h70mppx1mu.001rskips50.pkl", "rb") as f:
         trajectories = pickle.load(f)
     transitions = rollout.flatten_trajectories(trajectories)
     train(transitions)