ColossalAI/colossalai/utils/checkpoint/__init__.py

mirror of https://github.com/hpcaitech/ColossalAI
from .module_checkpoint import save_checkpoint, load_checkpoint

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

	`__all__ = ['save_checkpoint', 'load_checkpoint']`