From 31bf02ab95b85394fc172ef816eea59d86e93c13 Mon Sep 17 00:00:00 2001 From: Nick Walton Date: Thu, 14 Nov 2019 15:40:14 -0700 Subject: [PATCH] update --- 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 d32b533..81931d3 100644 --- a/generator/simple/finetune.py +++ b/generator/simple/finetune.py @@ -14,7 +14,7 @@ sess = gpt2.start_tf_sess() gpt2.finetune(sess, file_name, multi_gpu=True, - batch_size=16, + batch_size=8, learning_rate=0.0001, model_name=model_name, steps=4000)