WIP XLM + refactoring

2019-07-03 22:54:39 +02:00
parent 288be7b7ea
commit c41f2bad69
27 changed files with 515 additions and 136 deletions
--- a/hubconfs/bert_hubconf.py
+++ b/hubconfs/bert_hubconf.py
@@ -1,5 +1,5 @@
-from pytorch_pretrained_bert.tokenization import BertTokenizer
-from pytorch_pretrained_bert.modeling import (
+from pytorch_pretrained_bert.tokenization_bert import BertTokenizer
+from pytorch_pretrained_bert.modeling_bert import (
        BertModel,
        BertForNextSentencePrediction,
        BertForMaskedLM,