Extending HuggingFace's transformers package with new models.
Many of HuggingFace's finetuning models have redundant code.
These patterns have been refactored into Mixins and new
generalized training Heads allowing for more flexibility.
-
BERT
- BertForSequenceClassification
- BertForTokenClassification
-
RoBERTA
- RobertaForSequenceClassification
- RobertaForTokenClassification
- BERT
- BertForTokenCrfClassification