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

github/ColossalAI

mirror of https://github.com/hpcaitech/ColossalAI.git synced 2026-02-02 15:39:04 +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