ColossalAI/colossalai/checkpoint_io
wukong1992 6b305a99d6
[booster] torch fsdp fix ckpt (#3788)
2023-05-23 16:58:45 +08:00
..
__init__.py [checkpoint] refactored the API and added safetensors support (#3427) 2023-04-04 15:23:01 +08:00
checkpoint_io_base.py [plugin] torch ddp plugin supports sharded model checkpoint (#3775) 2023-05-18 20:05:59 +08:00
general_checkpoint_io.py [booster] torch fsdp fix ckpt (#3788) 2023-05-23 16:58:45 +08:00
index_file.py [NFC] fix typo applications/ and colossalai/ (#3735) 2023-05-15 11:46:25 +08:00
utils.py [plugin] torch ddp plugin supports sharded model checkpoint (#3775) 2023-05-18 20:05:59 +08:00