Files
ColossalAI/colossalai/utils/checkpoint/__init__.py
2022-07-06 17:22:03 +08:00

4 lines
114 B
Python

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