Changing name

asyml · Dec 26, 2019 · 824d947 · 824d947
1 parent 31d3e07
commit 824d947
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/texar/torch/modules/pretrained/bert.py b/texar/torch/modules/pretrained/bert.py
@@ -166,9 +166,9 @@ class PretrainedBERTMixin(PretrainedMixin, ABC):
             _SCIBERT_PATH + 'scibert_basevocab_cased.tar.gz',
 
         # BERT for MS-MARCO
-        'bert-msmarco-base':
+        'bert-msmarco-nogueira19-base':
             _BERT_MSMARCO_NOGUEIRA19_PATH + '1cyUrhs7JaCJTTu-DjFUqP6Bs4f8a6JTX/',
-        'bert-msmarco-large':
+        'bert-msmarco-nogueira19-large':
             _BERT_MSMARCO_NOGUEIRA19_PATH + '1crlASTMlsihALlkabAQP6JTYIZwC1Wm8/'
     }
     _MODEL2CKPT = {