File size: 291 Bytes
5fa1a76 |
1 2 3 4 5 6 7 8 9 10 11 |
from transformers import Seq2SeqTrainer trainer = Seq2SeqTrainer( args=training_args, model=model, train_dataset=dataset["train"], eval_dataset=dataset["test"], data_collator=data_collator, tokenizer=processor, ) And with that, you're ready to start training! |