Troubleshoot KeyError: loss
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!
2023-01-12T02:39:56Z
image766×60 2.95 KBThis could be why it’s not computing a loss, I just don’t know what to change.
2023-01-12T02:41:00Z
2023-01-12T02:41:24Z
image759×170 5.74 KB