ALBERT for SQuAD

2019-11-06 20:47:42 +00:00
parent 70d99980de
commit 4374eaea78
3 changed files with 100 additions and 6 deletions
--- a/transformers/init.py
+++ b/transformers/init.py
@@ -108,6 +108,7 @@ if is_torch_available():
    from .modeling_encoder_decoder import PreTrainedEncoderDecoder, Model2Model

    from .modeling_albert import (AlbertModel, AlbertForMaskedLM, AlbertForSequenceClassification,
+                                AlbertForQuestionAnswering,
                                load_tf_weights_in_albert, ALBERT_PRETRAINED_MODEL_ARCHIVE_MAP)

    # Optimization