ColossalAI

Author	SHA1	Message	Date
ver217	62b0a8d644	[zero] sharded optim support hybrid cpu adam (#486 ) * sharded optim support hybrid cpu adam * update unit test * polish docstring	2022-03-22 14:56:59 +08:00
Jiarui Fang	0fcfb1e00d	[test] make zero engine test really work (#447 )	2022-03-17 17:24:25 +08:00
Jiarui Fang	237d08e7ee	[zero] hybrid cpu adam (#445 )	2022-03-17 15:05:41 +08:00
Kai Wang (Victor Kai)	53bb3bcc0a	fix format (#362 )	2022-03-11 15:50:28 +08:00
LuGY	a3269de5c9	[zero] cpu adam kernel (#288 ) * Added CPU Adam * finished the cpu adam * updated the license * delete useless parameters, removed resnet * modified the method off cpu adam unittest * deleted some useless codes * removed useless codes Co-authored-by: ver217 <lhx0217@gmail.com> Co-authored-by: Frank Lee <somerlee.9@gmail.com> Co-authored-by: jiaruifang <fangjiarui123@gmail.com>	2022-03-11 15:50:28 +08:00