Adding new pretrained model to the help of the bert_model argument.

This commit is contained in:
Grégory Châtel committed 2019-01-02 14:00:59 +01:00
1 parent 8da280ebbe
commit 186f75342e
3 files changed
+6 -3

No files matched your search

+2 -1
View File
@@ -681,7 +681,8 @@ def main():
## Required parameters
parser.add_argument("--bert_model", default=None, type=str, required=True,
help="Bert pre-trained model selected in the list: bert-base-uncased, "
"bert-large-uncased, bert-base-cased, bert-base-multilingual, bert-base-chinese.")
"bert-large-uncased, bert-base-cased, bert-large-cased, bert-base-multilingual-uncased, "
"bert-base-multilingual-cased, bert-base-chinese.")
parser.add_argument("--output_dir", default=None, type=str, required=True,
help="The output directory where the model checkpoints and predictions will be written.")