ColossalAI/colossalai/utils/checkpoint
Jiarui Fang 9e4c6449b0
[checkpoint] add ColoOptimizer checkpointing (#1316)
2022-07-15 09:52:55 +08:00
..
__init__.py [checkpoint] checkpoint for ColoTensor Model (#1196) 2022-07-06 17:22:03 +08:00
module_checkpoint.py [checkpoint] add ColoOptimizer checkpointing (#1316) 2022-07-15 09:52:55 +08:00