ColossalAI/colossalai/utils/checkpoint
Jiarui Fang 3ef3791a3b
[checkpoint] add test for bert and hotfix save bugs (#1297)
2022-07-14 15:38:18 +08:00
..
__init__.py [checkpoint] checkpoint for ColoTensor Model (#1196) 2022-07-06 17:22:03 +08:00
module_checkpoint.py [checkpoint] add test for bert and hotfix save bugs (#1297) 2022-07-14 15:38:18 +08:00