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

[ahsannawazch]

worked for me too. Don’t know why

[shrijayan]

After reviewing the data_processor.py file, I can see the current data processing method. The prepare_dataset method currently creates a dataset with “input_ids”, “attention_mask”, and “labels” columns.