Hongxin Liu 5452df63c5 [plugin] torch ddp plugin supports sharded model checkpoint (#3775) ... * [plugin] torch ddp plugin add save sharded model * [test] fix torch ddp ckpt io test * [test] fix torch ddp ckpt io test * [test] fix low level zero plugin test * [test] fix low level zero plugin test * [test] add debug info * [test] add debug info * [test] add debug info * [test] add debug info * [test] add debug info * [test] fix low level zero plugin test * [test] fix low level zero plugin test * [test] remove debug info		2023-05-18 20:05:59 +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] gemini plugin support shard checkpoint (#3610)	2023-05-05 14:37:21 +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