update finetune

2026-09-09 11:13:26 +08:00 · 2019-11-14 12:52:15 -07:00
parent 5132fb5bbd
commit 7f07f55694
1 changed files with 3 additions and 0 deletions
@@ -13,6 +13,9 @@ file_name = "merged-first-person.txt"
 sess = gpt2.start_tf_sess()
 gpt2.finetune(sess,
              file_name,
+              multi_gpu=True,
+              batch_size=16,
+              sample_length=256,
              model_name=model_name,
              steps=10000)