preparing for first release

2018-11-15 20:56:10 +01:00
parent 8513741b57
commit 1de35b624b
21 changed files with 3045 additions and 958 deletions
--- a/pytorch_pretrained_bert/init.py
+++ b/pytorch_pretrained_bert/init.py
@@ -0,0 +1,5 @@
+from .tokenization import BertTokenizer, BasicTokenizer, WordpieceTokenizer
+from .modeling import (BertConfig, BertModel, BertForPreTraining,
+                       BertForMaskedLM, BertForNextSentencePrediction,
+                       BertForSequenceClassification, BertForQuestionAnswering)
+from .optimization import BERTAdam