diff --git a/generator/simple/finetune.py b/generator/simple/finetune.py index 402c6a7..cd472a6 100644 --- a/generator/simple/finetune.py +++ b/generator/simple/finetune.py @@ -15,7 +15,6 @@ gpt2.finetune(sess, file_name, multi_gpu=True, batch_size=16, - sample_length=256, learning_rate=0.0001, model_name=model_name, steps=1000)