Hi everyone, I want to do text classification using T5, but I encountered an error during training. Here is my code: from transformers import T5Tokenizer, T5ForConditionalGeneration, Trainer, TrainingArguments from datasets import load_dataset # Load the dataset dataset = load_dataset("dair-ai/em…

It worked. I don’t know if the result is correct… # Define the Trainer trainer = Trainer( model=model, args=training_args, train_dataset=tokenized_train_dataset, eval_dataset=tokenized_validation_dataset, #tokenizer=tokenizer, # <= this! )

ValueError when fine-tuning T5 for text classification

Beginners

system Closed February 12, 2025, 2:13am 7

This topic was automatically closed 12 hours after the last reply. New replies are no longer allowed.

Topic		Replies	Views
Help addapting pytorch/text-classification example to t5 🤗Transformers	4	1249	May 25, 2023
Fine-Tuning T5v1.1 Using Trainer API Beginners	0	514	November 4, 2021
Errors when fine-tuning T5 Beginners	7	6521	January 3, 2022
Finetuning T5 on custom data Models	0	1063	November 13, 2020
Finetuning Transformers for Text Classification Issue 🤗Transformers	2	710	May 11, 2023

ValueError when fine-tuning T5 for text classification

Related topics