Hi, I was going through the documentation and got a confusion trainer = Trainer( model=model, # the instantiated :hugs: Transformers model to be trained args=training_args, # training arguments, defined above train_dataset=train_dataset, # trainin…

Type of dataset in Trainer class

valhalla July 18, 2020, 9:23am 3

Hi @suyash21 this post has some explanation about the dataset expected by Trainer

1 Like

Topic		Replies	Views
Is Eval and Validation same in Trainer API? Beginners	4	1222	September 14, 2021
Pytorch trainer giving worse results than tensorflow Beginners	0	468	January 5, 2023
Evaluating huggingface transformer with trainer gives different results 🤗Transformers	0	656	March 22, 2023
Confusion about trainer.predict(dataset['test']) output 🤗Transformers	0	418	November 3, 2022
Difference between model.train and trainer.train Beginners	0	285	February 26, 2023