ColossalAI

Commit Graph

Author	SHA1	Message	Date
LuGY	6a3f9fda83	[cuda] modify the fused adam, support hybrid of fp16 and fp32 (#497 )	3 years ago
Jiarui Fang	5d7dc3525b	[hotfix] run cpu adam unittest in pytest (#424 )	3 years ago
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>	3 years ago

Author

SHA1

Message

Date

LuGY

6a3f9fda83

[cuda] modify the fused adam, support hybrid of fp16 and fp32 (#497 )

Jiarui Fang

5d7dc3525b

[hotfix] run cpu adam unittest in pytest (#424 )

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>

3 Commits (8d8c5407c0f677f9f356a96c2bef7dad83af12dd)