mirror of
https://github.com/wassname/Clover-Edition.git
synced 2026-10-29 12:40:16 +08:00
update finetune
This commit is contained in:
1 parent
5132fb5bbd
commit
7f07f55694
1 file changed
+3
@@ -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)
|
||||
|
||||
|
||||
Reference in new issue
Block a user