ColossalAI/colossalai/checkpoint_io/__init__.py at 1653063fce7062dfd652e73691fac44a24698cd5 - ColossalAI - Gitea: Git with a cup of tea at home

github/ColossalAI

mirror of https://github.com/hpcaitech/ColossalAI.git synced 2026-04-11 06:33:30 +00:00

Files

Frank Lee cd142fbefa [api] implemented the checkpoint io module (#3205 )

* [api] implemented the checkpoint io module

* polish code

* polish code

2023-03-23 10:53:17 +08:00

5 lines

205 B

Python

Raw Blame History

 from .checkpoint_io_base import CheckpointIO, ShardCheckpointIndexFile
 from .general_checkpoint_io import GeneralCheckpointIO
 __all__ = ['CheckpointIO', 'ShardCheckpointIndexFile', 'GeneralCheckpointIO']