File size: 252 Bytes
5fa1a76
 
 
 
1
2
3
4
from transformers import TFDistilBertForSequenceClassification
tf_model = TFDistilBertForSequenceClassification.from_pretrained("distilbert/distilbert-base-uncased")

Easily reuse this checkpoint for another task by switching to a different model head.