Update README.md
Browse files
README.md
CHANGED
@@ -28,6 +28,7 @@ metrics:
|
|
28 |
|
29 |
|
30 |
微调:
|
|
|
31 |
import numpy as np
|
32 |
from transformers import AutoModelForSeq2SeqLM, Seq2SeqTrainingArguments, Seq2SeqTrainer
|
33 |
import torch
|
@@ -71,9 +72,9 @@ metrics:
|
|
71 |
result["gen_len"] = np.mean(prediction_lens)
|
72 |
result = {k: round(v, 4) for k, v in result.items()}
|
73 |
return result
|
74 |
-
|
75 |
-
|
76 |
-
|
77 |
|
78 |
output_dir="./my_awesome_opus_books_model",
|
79 |
evaluation_strategy="epoch",
|
|
|
28 |
|
29 |
|
30 |
微调:
|
31 |
+
|
32 |
import numpy as np
|
33 |
from transformers import AutoModelForSeq2SeqLM, Seq2SeqTrainingArguments, Seq2SeqTrainer
|
34 |
import torch
|
|
|
72 |
result["gen_len"] = np.mean(prediction_lens)
|
73 |
result = {k: round(v, 4) for k, v in result.items()}
|
74 |
return result
|
75 |
+
model = AutoModelForSeq2SeqLM.from_pretrained(checkpoint)
|
76 |
+
batchsize=4
|
77 |
+
training_args = Seq2SeqTrainingArguments(
|
78 |
|
79 |
output_dir="./my_awesome_opus_books_model",
|
80 |
evaluation_strategy="epoch",
|