mirror of https://github.com/hpcaitech/ColossalAI
5 lines
205 B
Python
5 lines
205 B
Python
|
from .checkpoint_io_base import CheckpointIO, ShardCheckpointIndexFile
|
||
|
|
from .general_checkpoint_io import GeneralCheckpointIO
|
||
|
|
|
||
|
|
__all__ = ['CheckpointIO', 'ShardCheckpointIndexFile', 'GeneralCheckpointIO']
|