Skip to content

GitLab

"vscode:/vscode.git/clone" did not exist on "8d48c77b4dafb09450c50b404629c5ae0f33f7d6"
Commit 9b2540b5 authored by thomwolf's avatar thomwolf
Browse files

update __init__

parent bd3b3aee
Show whitespace changes
Inline Side-by-side
Showing with 6 additions and 6 deletions
pytorch_pretrained_bert/__init__.py
View file @ 9b2540b5
......@@ -6,15 +6,15 @@ from .tokenization_transfo_xl import (TransfoXLTokenizer, TransfoXLCorpus)
from .modeling import (BertConfig, BertModel, BertForPreTraining,
BertForMaskedLM, BertForNextSentencePrediction,
BertForSequenceClassification, BertForMultipleChoice,
BertForTokenClassification, BertForQuestionAnswering)
BertForTokenClassification, BertForQuestionAnswering,
load_tf_weights_in_bert)
from .modeling_openai import (OpenAIGPTConfig, OpenAIGPTModel,
OpenAIGPTLMHeadModel, OpenAIGPTDoubleHeadsModel)
from .modeling_transfo_xl import (TransfoXLConfig, TransfoXLModel)
OpenAIGPTLMHeadModel, OpenAIGPTDoubleHeadsModel,
load_tf_weights_in_openai_gpt)
from .modeling_transfo_xl import (TransfoXLConfig, TransfoXLModel,
load_tf_weights_in_transfo_xl)
from .optimization import BertAdam
from .optimization_openai import OpenAIAdam
from .convert_openai_checkpoint_to_pytorch import load_tf_weights_in_openai_gpt
from .convert_tf_checkpoint_to_pytorch import load_tf_weights_in_bert
from .convert_transfo_xl_checkpoint_to_pytorch import load_tf_weights_in_transfo_xl
from .file_utils import PYTORCH_PRETRAINED_BERT_CACHE
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment