From 4acba1d73d49c30d2d6aa2fddf7067e0283e74d1 Mon Sep 17 00:00:00 2001 From: Nick Walton Date: Thu, 14 Nov 2019 15:07:31 -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 cd472a6..d32b533 100644 --- a/generator/simple/finetune.py +++ b/generator/simple/finetune.py @@ -17,6 +17,6 @@ gpt2.finetune(sess, batch_size=16, learning_rate=0.0001, model_name=model_name, - steps=1000) + steps=4000) gpt2.generate(sess) \ No newline at end of file