update cache_dir in readme and examples

2018-11-26 10:45:13 +01:00
parent 63ae5d2134
commit 05053d163c
4 changed files with 8 additions and 5 deletions
--- a/pytorch_pretrained_bert/init.py
+++ b/pytorch_pretrained_bert/init.py
@@ -3,3 +3,4 @@ from .modeling import (BertConfig, BertModel, BertForPreTraining,
                       BertForMaskedLM, BertForNextSentencePrediction,
                       BertForSequenceClassification, BertForQuestionAnswering)
 from .optimization import BertAdam
+from .file_utils import PYTORCH_PRETRAINED_BERT_CACHE