ColossalAI/colossalai/utils/checkpoint
HELSON b6fd165f66
[checkpoint] add kwargs for load_state_dict (#1374)
2022-07-28 15:56:52 +08:00
..
__init__.py [checkpoint] checkpoint for ColoTensor Model (#1196) 2022-07-06 17:22:03 +08:00
module_checkpoint.py [checkpoint] add kwargs for load_state_dict (#1374) 2022-07-28 15:56:52 +08:00
utils.py [fx] add torchaudio test (#1369) 2022-07-27 11:03:14 +08:00