This commit is contained in:
Nick Walton
2019-11-14 15:40:14 -07:00
parent 4acba1d73d
commit 31bf02ab95
+1 -1
View File
@@ -14,7 +14,7 @@ sess = gpt2.start_tf_sess()
gpt2.finetune(sess,
file_name,
multi_gpu=True,
batch_size=16,
batch_size=8,
learning_rate=0.0001,
model_name=model_name,
steps=4000)