File size: 312 Bytes
5fa1a76 |
1 2 3 4 5 6 7 8 9 10 |
from transformers import DefaultDataCollator data_collator = DefaultDataCollator() </pt> <tf>py from transformers import DefaultDataCollator data_collator = DefaultDataCollator(return_tensors="tf") Train If you aren't familiar with finetuning a model with the [Trainer], take a look at the basic tutorial here! |