This commit is contained in:
Nick Walton
2019-11-14 15:07:31 -07:00
parent b0f7e296ac
commit 4acba1d73d
+1 -1
View File
@@ -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)