Merge pull request huggingface#47 from stevezheng23/dev/zheng/coqa

add mat-coqa runner with multitask + adversarial training support (co…
stevezheng23 · Nov 23, 2019 · 99a7a73 · 99a7a73
2 parents 93a2707 + 18df317
commit 99a7a73
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/transformers/__init__.py b/transformers/__init__.py
@@ -90,7 +90,8 @@
     from .modeling_roberta import (RobertaForMaskedLM, RobertaModel, RobertaModelAT,
                                    RobertaForSequenceClassification, RobertaForTokenClassification,
                                    RobertaForMultipleChoice, RobertaForQuestionAnswering, RobertaForQuestionAnsweringComplex,
-                                   RobertaForQuestionAnsweringKD, RobertaForQuestionAnsweringAT, RobertaForQuestionAnsweringMT,
+                                   RobertaForQuestionAnsweringKD, RobertaForQuestionAnsweringAT,
+                                   RobertaForQuestionAnsweringMT, RobertaForQuestionAnsweringMAT,
                                    ROBERTA_PRETRAINED_MODEL_ARCHIVE_MAP)
     from .modeling_distilbert import (DistilBertForMaskedLM, DistilBertModel,
                                 DistilBertForSequenceClassification, DistilBertForQuestionAnswering,