5fa1a76
1
2
3
4
from transformers import DistilBertForSequenceClassification model = DistilBertForSequenceClassification.from_pretrained("distilbert/distilbert-base-uncased") Easily reuse this checkpoint for another task by switching to a different model head.