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

Saugatkafley January 30, 2024, 4:12pm 13

Keeping the columns as text and label solved the issue.

1 Like

Topic		Replies	Views
ValueError: The model did not return a loss from the inputs, only the following keys: logits. For reference, the inputs it received are input_ids,attention_mask 🤗Transformers	1	1185	August 25, 2023
Multi_class_classification errors when fine-tuning via TrainerAPI Beginners	0	374	February 20, 2023
Chapter 3 questions Course	149	10515	August 29, 2025
Fine Tuning IMDb tutorial - Unable to reproduce and adapt Beginners	19	8600	August 21, 2020
I tired and can't solve this error , ValueError: The model did not return a loss from the inputs, only the following keys: logits. For reference, the inputs it received are input_ids,attention_mask Models	1	1160	March 29, 2023