Merge pull request #1792 from stefan-it/distilbert-for-token-classification

DistilBERT for token classification
2019-11-14 22:47:53 +01:00
parent 9629e2c676 05db5bc1af
commit 74ce8de7d8
5 changed files with 114 additions and 6 deletions
--- a/transformers/init.py
+++ b/transformers/init.py
@@ -94,6 +94,7 @@ if is_torch_available():
                                ROBERTA_PRETRAINED_MODEL_ARCHIVE_MAP)
    from .modeling_distilbert import (DistilBertForMaskedLM, DistilBertModel,
                                DistilBertForSequenceClassification, DistilBertForQuestionAnswering,
+                                DistilBertForTokenClassification,
                                DISTILBERT_PRETRAINED_MODEL_ARCHIVE_MAP)
    from .modeling_encoder_decoder import PreTrainedEncoderDecoder, Model2Model