__init__.py 455 Bytes
Newer Older
1
__version__ = "0.4.0"
thomwolf's avatar
thomwolf committed
2
3
4
from .tokenization import BertTokenizer, BasicTokenizer, WordpieceTokenizer
from .modeling import (BertConfig, BertModel, BertForPreTraining,
                       BertForMaskedLM, BertForNextSentencePrediction,
thomwolf's avatar
thomwolf committed
5
6
                       BertForSequenceClassification, BertForTokenClassification,
                       BertForQuestionAnswering)
thomwolf's avatar
thomwolf committed
7
from .optimization import BertAdam
8
from .file_utils import PYTORCH_PRETRAINED_BERT_CACHE