Troubleshoot KeyError: loss

I get the KeyError: loss when running trainer.train()

Here’s what my dataset looks like:
image

That’s a dataframe, which I convert to a dataset for the trainer.

Fine-tuning distilgpt2. I noticed the model doesn’t take “lables” as a param. Did I configure it right.

Any guidance on where to look is much appreciated.

Thank you!


This could be why it’s not computing a loss, I just don’t know what to change.

image