"tests/test_modeling_tf_xlm_roberta.py" did not exist on "e392ba6938f50655a195ea7ec8a260b1e9fc6058"
__init__.py 716 Bytes
Newer Older
thomwolf's avatar
thomwolf committed
1
__version__ = "0.5.0"
thomwolf's avatar
thomwolf committed
2
from .tokenization import BertTokenizer, BasicTokenizer, WordpieceTokenizer
thomwolf's avatar
thomwolf committed
3
from .tokenization_openai import OpenAIGPTTokenizer
thomwolf's avatar
thomwolf committed
4
5
from .modeling import (BertConfig, BertModel, BertForPreTraining,
                       BertForMaskedLM, BertForNextSentencePrediction,
6
7
                       BertForSequenceClassification, BertForMultipleChoice,
                       BertForTokenClassification, BertForQuestionAnswering)
thomwolf's avatar
thomwolf committed
8
9
from .modeling_openai import (OpenAIGPTConfig, OpenAIGPTModel,
                              OpenAIGPTLMHeadModel, OpenAIGPTDoubleHeadsModel)
thomwolf's avatar
thomwolf committed
10
from .optimization import BertAdam
thomwolf's avatar
thomwolf committed
11
from .optimization_openai import OpenAIAdam
12
from .file_utils import PYTORCH_PRETRAINED_BERT_CACHE