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.
ColossalAI/colossalai/utils/checkpoint/__init__.py

4 lines
114 B

from .module_checkpoint import save_checkpoint, load_checkpoint
__all__ = ['save_checkpoint', 'load_checkpoint']