ColossalAI/colossalai/utils/checkpoint
Frank Lee 0c1a16ea5b
[util] standard checkpoint function naming (#1377)
2022-07-28 09:29:30 +08:00
..
__init__.py [checkpoint] checkpoint for ColoTensor Model (#1196) 2022-07-06 17:22:03 +08:00
module_checkpoint.py [util] standard checkpoint function naming (#1377) 2022-07-28 09:29:30 +08:00
utils.py [fx] add torchaudio test (#1369) 2022-07-27 11:03:14 +08:00