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
..
mixed_precision	[amp] Add naive amp demo (#3774)	2023-05-18 16:33:14 +08:00
plugin	[plugin] torch ddp plugin supports sharded model checkpoint (#3775)	2023-05-18 20:05:59 +08:00
__init__.py	[booster] implemented the torch ddd + resnet example (#3232)	2023-03-27 10:24:14 +08:00
accelerator.py	[booster] added the accelerator implementation (#3159)	2023-03-20 13:59:24 +08:00
booster.py	[doc] Fix typo under colossalai and doc(#3618)	2023-04-26 11:38:43 +08:00