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