From a124587741de80a694da95c4db5be5d4c42b59e0 Mon Sep 17 00:00:00 2001
From: Nick Walton <nickwalton00@gmail.com>
Date: Thu, 14 Nov 2019 13:52:11 -0700
Subject: [PATCH] update train params

---
 generator/simple/finetune.py | 5 +++--
 1 file changed, 3 insertions(+), 2 deletions(-)

diff --git a/generator/simple/finetune.py b/generator/simple/finetune.py
index ad5cdcc..481438c 100644
--- a/generator/simple/finetune.py
+++ b/generator/simple/finetune.py
@@ -14,9 +14,10 @@ sess = gpt2.start_tf_sess()
 gpt2.finetune(sess,
               file_name,
               multi_gpu=True,
-              batch_size=16,
+              batch_size=4,
               sample_length=256,
+              learning_rate=0.0005,
               model_name=model_name,
-              steps=10000)
+              steps=1000)
 
 gpt2.generate(sess)
\ No newline at end of file