add tf auto models + tests

This commit is contained in:
thomwolf
2019-09-05 12:21:08 +02:00
parent f5c698b21a
commit 01597e5b90
4 changed files with 581 additions and 12 deletions

View File

@@ -96,8 +96,15 @@ if _tf_available:
logger.info("TensorFlow version {} available.".format(tf.__version__))
from .modeling_tf_utils import TFPreTrainedModel
from .modeling_tf_auto import (TFAutoModel, TFAutoModelForSequenceClassification, TFAutoModelForQuestionAnswering,
TFAutoModelWithLMHead)
from .modeling_tf_bert import (TFBertPreTrainedModel, TFBertModel, TFBertForPreTraining,
TFBertForMaskedLM, TFBertForNextSentencePrediction, load_bert_pt_weights_in_tf)
TFBertForMaskedLM, TFBertForNextSentencePrediction,
TFBertForSequenceClassification, TFBertForMultipleChoice,
TFBertForTokenClassification, TFBertForQuestionAnswering,
load_bert_pt_weights_in_tf,
TF_BERT_PRETRAINED_MODEL_ARCHIVE_MAP)
# Files and general utilities