From 6e0de2f72a92973a983ab181fb395987e175304f Mon Sep 17 00:00:00 2001
From: Nick Walton <nickwalton00@gmail.com>
Date: Thu, 14 Nov 2019 14:02:37 -0700
Subject: [PATCH] update train params

---
 generator/simple/finetune.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/generator/simple/finetune.py b/generator/simple/finetune.py
index 481438c..2f0ada0 100644
--- a/generator/simple/finetune.py
+++ b/generator/simple/finetune.py
@@ -16,7 +16,7 @@ gpt2.finetune(sess,
               multi_gpu=True,
               batch_size=4,
               sample_length=256,
-              learning_rate=0.0005,
+              learning_rate=0.0002,
               model_name=model_name,
               steps=1000)