From bf67be1f14f0288471e367defb62598f889b5880 Mon Sep 17 00:00:00 2001 From: root Date: Sat, 30 Nov 2019 14:19:07 +0000 Subject: [PATCH] updatefinetune --- generator/simple/finetune.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/generator/simple/finetune.py b/generator/simple/finetune.py index 47334c4..7c0e995 100644 --- a/generator/simple/finetune.py +++ b/generator/simple/finetune.py @@ -8,18 +8,18 @@ if not os.path.isdir(os.path.join("models", model_name)): print("Downloading ", model_name, " model...") gpt2.download_gpt2(model_name=model_name) # model is saved into current directory under /models/124M/ -file_name = "writing_prompts.txt" +file_name = "text_adventures.txt" sess = gpt2.start_tf_sess() gpt2.finetune(sess, file_name, multi_gpu=True, - batch_size=8, + batch_size=32, learning_rate=0.0001, model_name=model_name, sample_every=1000, - max_checkpoints=1, + max_checkpoints=8, save_every=200, - steps=600) + steps=1000) gpt2.generate(sess)