support latest multi language bert fine tune

fix issue of bert-base-multilingual and add support for uncased multilingual
This commit is contained in:
Ahmad Barqawi 2019-05-27 11:27:41 +02:00
parent 3fc63f126d
commit c4fe56dcc0

View File

@ -235,7 +235,7 @@ def main():
parser.add_argument("--output_dir", type=Path, required=True)
parser.add_argument("--bert_model", type=str, required=True,
choices=["bert-base-uncased", "bert-large-uncased", "bert-base-cased",
"bert-base-multilingual", "bert-base-chinese"])
"bert-base-multilingual-uncased", "bert-base-chinese", "bert-base-multilingual-cased"])
parser.add_argument("--do_lower_case", action="store_true")
parser.add_argument("--reduce_memory", action="store_true",