converting distilled/fine-tuned models

2019-09-24 10:58:52 +02:00
parent 8ba44ced95
commit a7e01a248b
4 changed files with 90 additions and 39 deletions
--- a/pytorch_transformers/init.py
+++ b/pytorch_transformers/init.py
@@ -146,7 +146,7 @@ if _tf_available:
    from .modeling_tf_distilbert import (TFDistilBertPreTrainedModel, TFDistilBertMainLayer,
                                         TFDistilBertModel, TFDistilBertForMaskedLM,
                                         TFDistilBertForSequenceClassification,
-                                         TFDistilBertForSequenceClassification,
+                                         TFDistilBertForQuestionAnswering,
                                         load_distilbert_pt_weights_in_tf2,
                                         TF_DISTILBERT_PRETRAINED_MODEL_ARCHIVE_MAP)