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.