File size: 295 Bytes
5fa1a76 |
1 2 3 4 5 6 7 8 |
Load T5 with [AutoModelForSeq2SeqLM]: from transformers import AutoModelForSeq2SeqLM, Seq2SeqTrainingArguments, Seq2SeqTrainer model = AutoModelForSeq2SeqLM.from_pretrained(checkpoint) At this point, only three steps remain: Define your training hyperparameters in [Seq2SeqTrainingArguments]. |