diff --git a/generator/simple/finetune.py b/generator/simple/finetune.py index 7c0e995..2c599e8 100644 --- a/generator/simple/finetune.py +++ b/generator/simple/finetune.py @@ -17,7 +17,7 @@ gpt2.finetune(sess, batch_size=32, learning_rate=0.0001, model_name=model_name, - sample_every=1000, + sample_every=10000, max_checkpoints=8, save_every=200, steps=1000)