This is because DataCollatorWithPadding returns a dict with labels as its elements. If you want to handle other data appropriately, you will need to write your own collate_fn (Data Collator).
1 Like
This is because DataCollatorWithPadding returns a dict with labels as its elements. If you want to handle other data appropriately, you will need to write your own collate_fn (Data Collator).