The model did not return a loss from the inputs, only the following keys: logits. For reference, the inputs it received are input_values

Rename columns to text and labels for text classification using distilbert-base-uncased model. It needs to be checked for other domains and models too.

17 Likes