Tests for ALBERT in TF2 + fixes

2019-11-08 00:12:21 +00:00
parent 7bddbf5961
commit b18509c208
3 changed files with 237 additions and 11 deletions
--- a/transformers/init.py
+++ b/transformers/init.py
@@ -169,6 +169,7 @@ if is_tf_available():
                                    TF_CTRL_PRETRAINED_MODEL_ARCHIVE_MAP)

    from .modeling_tf_albert import (TFAlbertPreTrainedModel, TFAlbertModel, TFAlbertForMaskedLM,
+                                     TFAlbertForSequenceClassification,
                                    TF_ALBERT_PRETRAINED_MODEL_ARCHIVE_MAP)

 # TF 2.0 <=> PyTorch conversion utilities