mirror of
https://github.com/wassname/Clover-Edition.git
synced 2026-10-29 12:40:16 +08:00
update train params
This commit is contained in:
1 parent
7f07f55694
commit
a124587741
1 file changed
+3
-2
@@ -14,9 +14,10 @@ sess = gpt2.start_tf_sess()
|
||||
gpt2.finetune(sess,
|
||||
file_name,
|
||||
multi_gpu=True,
|
||||
batch_size=16,
|
||||
batch_size=4,
|
||||
sample_length=256,
|
||||
learning_rate=0.0005,
|
||||
model_name=model_name,
|
||||
steps=10000)
|
||||
steps=1000)
|
||||
|
||||
gpt2.generate(sess)
|
||||
Reference in new issue
Block a user