mirror of
https://github.com/hpcaitech/ColossalAI.git
synced 2025-08-19 00:17:18 +00:00
Merge 105d9745b1
into 46ed5d856b
This commit is contained in:
commit
4007888ab2
@ -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
|
||||
|
||||
|
Loading…
Reference in New Issue
Block a user