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

2019-01-02 14:00:59 +01:00
parent 8da280ebbe
commit 186f75342e
3 changed files with 6 additions and 3 deletions
--- 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,