The following columns in the training set don't have a corresponding argument

Hi,

The Trainer automatically ignores columns in your dataset which aren’t used by the model. For T5 for instance, the model expects input_ids, attention_mask, labels etc., but not “summary”, “document”, “id”. As long as input_ids etc are in your dataset, it’s fine.

The warning is just telling you that those columns aren’t used.

2 Likes