I get the KeyError: loss when running trainer.train()
Here’s what my dataset looks like:
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!