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
..
bookkeeping	Feature/zero (#279)	2022-03-11 15:50:28 +08:00
__init__.py	impl shard optim v2 and add unit test	2022-03-11 15:50:28 +08:00
_utils.py	Feature/zero (#279)	2022-03-11 15:50:28 +08:00
sharded_optim.py	[zero] cpu adam kernel (#288)	2022-03-11 15:50:28 +08:00
sharded_optim_v2.py	polish code	2022-03-11 15:50:28 +08:00