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! |