improvement of regexp (including support of unicode)

This commit is contained in:
igeni 2024-03-21 22:05:04 +03:00
parent 848a574c26
commit 105d9745b1

View File

@ -35,7 +35,7 @@ def validate_case_matches_checkpoint(do_lower_case, init_checkpoint):
if not init_checkpoint:
return
m = re.match("^.*?([A-Za-z0-9_-]+)/bert_model.ckpt", init_checkpoint)
m = re.match("^.*?([\w-]+)/bert_model.ckpt", init_checkpoint)
if m is None:
return