@valhalla I read through that post and I see how the OP would get that error. However, I’m still not clear about creating the DataSet or DataLoader object that is expected. My guess is I need to create a custom DataSet object. I’m reading through this tutorial: https://pytorch.org/tutorials/beginner/data_loading_tutorial.html# 3
For context, my data is setup in a csv file. The first column has name “Title” and consists of the titles of various scientific abstracts. The second column has name “Raw” and is the raw text from those scientific abstracts. Here’s an example:
Title, Raw
DistillBERT a distilled version of BERT smaller faster cheaper and lighter, As transfer learning from large-scale pretrained models becomes more prevalent in Natural Language Processing (NLP) blah blah blah.

It seems creating a custom DataSet object wouldn’t be too difficult, but I’m not sure if that’s the best approach.