/
githubmirror
/
ColossalAI
Обзор
Документация
Войти
/
githubmirror
/
ColossalAI
Код
Запросы
0
Пакеты
0
Релизы
0
Аналитика
Безопасность
main
colossalai/checkpoint_io/__init__.py
13 строк
404 B
Haze188
[MoE/ZeRO] Moe refactor with zero refactor (#5821)
28 июн 2024, 09:00
Не верифицирован
28 июн 2024, 09:00
416580b
Код
Авторство
О чём код?
from .checkpoint_io_base import CheckpointIO from .general_checkpoint_io import GeneralCheckpointIO from .hybrid_parallel_checkpoint_io import HybridParallelCheckpointIO from .index_file import CheckpointIndexFile from .moe_checkpoint import MoECheckpointIO __all__ = [ "CheckpointIO", "CheckpointIndexFile", "GeneralCheckpointIO", "HybridParallelCheckpointIO", "MoECheckpointIO", ]