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