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

Intermediate
13

Keeping the columns as text and label solved the issue.

1 Like