mirror of https://github.com/hpcaitech/ColossalAI
You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
10 lines
253 B
10 lines
253 B
2 years ago
|
import re
|
||
|
|
|
||
|
|
GLOBAL_META_FILE_NAME = 'global_meta.bin'
|
||
|
|
MODEL_CKPT_FILE_NAME = 'model.bin'
|
||
|
|
OPTIM_CKPT_FILE_NAME = 'optim.bin'
|
||
|
|
META_CKPT_FILE_NAME = 'meta.bin'
|
||
|
|
OTHER_CKPT_FILE_NAME = 'other.bin'
|
||
|
|
|
||
|
|
CKPT_PAT = re.compile(r'global_meta|model|optim|meta|other')
|