What is the desired input columns? isn't "input_ids", "labels" and "attention_mask"?
#135
by
lserlohn
- opened
I created my model, and got the following error. I could't figure out where is the problem.
ValueError: No columns in the dataset match the model's forward method signature. The following columns have been ignored: [attention_mask, input_ids, labels]. Please check the dataset and model. You may need to set remove_unused_columns=False in TrainingArguments.