mirror of
https://github.com/hpcaitech/ColossalAI.git
synced 2025-07-25 20:46:13 +00:00
improvement of regexp (including support of unicode)
This commit is contained in:
parent
848a574c26
commit
105d9745b1
@ -35,7 +35,7 @@ def validate_case_matches_checkpoint(do_lower_case, init_checkpoint):
|
|||||||
if not init_checkpoint:
|
if not init_checkpoint:
|
||||||
return
|
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:
|
if m is None:
|
||||||
return
|
return
|
||||||
|
|
||||||
|
Loading…
Reference in New Issue
Block a user