update train params

2026-09-11 11:51:53 +08:00 · 2019-11-14 14:02:37 -07:00
parent a124587741
commit 6e0de2f72a
1 changed files with 1 additions and 1 deletions
@@ -16,7 +16,7 @@ gpt2.finetune(sess,
              multi_gpu=True,
              batch_size=4,
              sample_length=256,
-              learning_rate=0.0005,
+              learning_rate=0.0002,
              model_name=model_name,
              steps=1000)