File size: 203 Bytes
5fa1a76
 
 
 
 
1
2
3
4
5
Remove any columns you don't need:

tokenized_squad = squad.map(preprocess_function, batched=True, remove_columns=squad["train"].column_names)

Now create a batch of examples using [DefaultDataCollator].