ColossalAI/colossalai/utils/checkpoint at c92f84fcdb772a3f602185c1be4db0cf27b5aa75 - ColossalAI - Gitea: Git with a cup of tea at home

github/ColossalAI

mirror of https://github.com/hpcaitech/ColossalAI.git synced 2026-01-18 16:46:08 +00:00

Files

History

Jiarui Fang c92f84fcdb [tensor] distributed checkpointing for parameters (#1240 )

2022-07-12 15:51:06 +08:00

..

__init__.py

[checkpoint] checkpoint for ColoTensor Model (#1196 )

2022-07-06 17:22:03 +08:00

module_checkpoint.py

[tensor] distributed checkpointing for parameters (#1240 )

2022-07-12 15:51:06 +08:00