DDDSSS commited on
Commit
61b749b
1 Parent(s): 2da6076

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +4 -3
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
- model = AutoModelForSeq2SeqLM.from_pretrained(checkpoint)
75
- batchsize=4
76
- training_args = Seq2SeqTrainingArguments(
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",