Skip to content

Commit e61b57e

Browse files
author
Atif Ahmed
committed
Changing name
1 parent 25739de commit e61b57e

File tree

2 files changed

+11
-11
lines changed

2 files changed

+11
-11
lines changed

texar/torch/data/tokenizers/bert_tokenizer.py

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -76,8 +76,8 @@ class BERTTokenizer(PretrainedBERTMixin, TokenizerBase):
7676
'scibert-basevocab-cased': 512,
7777

7878
# BERT for MS-MARCO
79-
'bert-msmarco-base': 512,
80-
'bert-msmarco-large': 512,
79+
'bert-msmarco-nogueira19-base': 512,
80+
'bert-msmarco-nogueira19-large': 512,
8181
}
8282
_VOCAB_FILE_NAMES = {'vocab_file': 'vocab.txt'}
8383
_VOCAB_FILE_MAP = {
@@ -104,8 +104,8 @@ class BERTTokenizer(PretrainedBERTMixin, TokenizerBase):
104104
'scibert-basevocab-cased': 'vocab.txt',
105105

106106
# BERT for MS-MARCO
107-
'bert-msmarco-base': 'vocab.txt',
108-
'bert-msmarco-large': 'vocab.txt',
107+
'bert-msmarco-nogueira19-base': 'vocab.txt',
108+
'bert-msmarco-nogueira19-large': 'vocab.txt',
109109
}
110110
}
111111

texar/torch/modules/pretrained/bert.py

Lines changed: 7 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -34,7 +34,7 @@
3434
_BIOBERT_PATH = "https://github.com/naver/biobert-pretrained/releases/download/"
3535
_SCIBERT_PATH = "https://s3-us-west-2.amazonaws.com/ai2-s2-research/" \
3636
"scibert/tensorflow_models/"
37-
_BERT_MSMARCO_PATH = "https://drive.google.com/file/d/"
37+
_BERT_MSMARCO_NOGUEIRA19_PATH = "https://drive.google.com/file/d/"
3838

3939

4040
class PretrainedBERTMixin(PretrainedMixin, ABC):
@@ -103,9 +103,9 @@ class PretrainedBERTMixin(PretrainedMixin, ABC):
103103
(`Nguyen et al`., 2016) dataset. It's the best performing model (on Jan
104104
8th 2019) on MS-MARCO Passage re-ranking task. Two models are included:
105105
106-
* ``bert-msmarco-base``: Original BERT base model fine-tuned on
106+
* ``bert-msmarco-nogueira19-base``: Original BERT base model fine-tuned on
107107
MS-MARCO.
108-
* ``bert-msmarco-large``: Original BERT large model fine-tuned on
108+
* ``bert-msmarco-nogueira19-large``: Original BERT large model fine-tuned on
109109
MS-MARCO.
110110
111111
We provide the following BERT classes:
@@ -167,9 +167,9 @@ class PretrainedBERTMixin(PretrainedMixin, ABC):
167167

168168
# BERT for MS-MARCO
169169
'bert-msmarco-base':
170-
_BERT_MSMARCO_PATH + '1cyUrhs7JaCJTTu-DjFUqP6Bs4f8a6JTX/',
170+
_BERT_MSMARCO_NOGUEIRA19_PATH + '1cyUrhs7JaCJTTu-DjFUqP6Bs4f8a6JTX/',
171171
'bert-msmarco-large':
172-
_BERT_MSMARCO_PATH + '1crlASTMlsihALlkabAQP6JTYIZwC1Wm8/'
172+
_BERT_MSMARCO_NOGUEIRA19_PATH + '1crlASTMlsihALlkabAQP6JTYIZwC1Wm8/'
173173
}
174174
_MODEL2CKPT = {
175175
# Standard BERT
@@ -194,8 +194,8 @@ class PretrainedBERTMixin(PretrainedMixin, ABC):
194194
'scibert-basevocab-cased': 'bert_model.ckpt',
195195

196196
# BERT for MSMARCO
197-
'bert-msmarco-base': 'model.ckpt-100000',
198-
'bert-msmarco-large': 'model.ckpt-100000',
197+
'bert-msmarco-nogueira19-base': 'model.ckpt-100000',
198+
'bert-msmarco-nogueira19-large': 'model.ckpt-100000',
199199
}
200200

201201
@classmethod

0 commit comments

Comments
 (0)