Update README.md
Browse files
README.md
CHANGED
@@ -32,6 +32,16 @@ More information needed
|
|
32 |
More information needed
|
33 |
|
34 |
## Training procedure
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
35 |
|
36 |
### Training hyperparameters
|
37 |
|
|
|
32 |
More information needed
|
33 |
|
34 |
## Training procedure
|
35 |
+
from transformers import Seq2SeqTrainer
|
36 |
+
|
37 |
+
trainer = Seq2SeqTrainer(
|
38 |
+
args=training_args,
|
39 |
+
model=model,
|
40 |
+
train_dataset=dataset["train"],
|
41 |
+
eval_dataset=dataset["test"],
|
42 |
+
data_collator=data_collator,
|
43 |
+
tokenizer=processor,
|
44 |
+
)
|
45 |
|
46 |
### Training hyperparameters
|
47 |
|