From 186f75342eed9f7bd2505b1b41ef317ea89d657b Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Gr=C3=A9gory=20Ch=C3=A2tel?= Date: Wed, 2 Jan 2019 14:00:59 +0100 Subject: [PATCH] Adding new pretrained model to the help of the `bert_model` argument. --- examples/run_classifier.py | 3 ++- examples/run_squad.py | 3 ++- examples/run_swag.py | 3 ++- 3 files changed, 6 insertions(+), 3 deletions(-) diff --git a/examples/run_classifier.py b/examples/run_classifier.py index adf81f4e28..e265ed73df 100644 --- a/examples/run_classifier.py +++ b/examples/run_classifier.py @@ -312,7 +312,8 @@ def main(): help="The input data dir. Should contain the .tsv files (or other data files) for the task.") 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("--task_name", default=None, type=str, diff --git a/examples/run_squad.py b/examples/run_squad.py index 6a97dd300b..8be4143a58 100644 --- a/examples/run_squad.py +++ b/examples/run_squad.py @@ -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.") diff --git a/examples/run_swag.py b/examples/run_swag.py index caddbee8ab..c31696fec8 100644 --- a/examples/run_swag.py +++ b/examples/run_swag.py @@ -249,7 +249,8 @@ def main(): help="The input data dir. Should contain the .csv files (or other data files) for the task.") 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,