Merge pull request #640 from Barqawiz/master

Support latest multi language bert fine tune
This commit is contained in:
Thomas Wolf authored and GitHub committed 2019-06-14 16:57:02 +02:00
commit 460d9afd45
1 file changed
+1 -1
@@ -270,7 +270,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("--do_whole_word_mask", action="store_true",
help="Whether to use whole word masking rather than per-WordPiece masking.")