ColossalAI/colossalai/utils/checkpoint
Jiarui Fang c92f84fcdb
[tensor] distributed checkpointing for parameters (#1240)
2022-07-12 15:51:06 +08:00
..
__init__.py
module_checkpoint.py [tensor] distributed checkpointing for parameters (#1240) 2022-07-12 15:51:06 +08:00