mirror of https://github.com/hpcaitech/ColossalAI
10 lines
253 B
Python
10 lines
253 B
Python
|
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')
|