The EvalPrediction object received in Trainers compute_metrics function returns an unequal number of label_ids and predictions.
My compute_metrics function given to the Trainer looks like this:

The shapes I print:

And the error message I get:

image807×55 8.67 KB

Does anyone of you have a solution to this?