Added test for AlbertForTokenClassification

2020-02-27 12:24:20 +01:00
parent aceb6a0907
commit f71157529e
1 changed files with 1 additions and 0 deletions
--- a/tests/test_modeling_albert.py
+++ b/tests/test_modeling_albert.py
@@ -29,6 +29,7 @@ if is_torch_available():
        AlbertModel,
        AlbertForMaskedLM,
        AlbertForSequenceClassification,
+        AlbertForTokenClassification,
        AlbertForQuestionAnswering,
    )
    from transformers.modeling_albert import ALBERT_PRETRAINED_MODEL_ARCHIVE_MAP