ColossalAI

Commit Graph

Author	SHA1	Message	Date
LuGY	6a3f9fda83	[cuda] modify the fused adam, support hybrid of fp16 and fp32 (#497 )	2022-03-25 14:15:53 +08:00
Jiarui Fang	5d7dc3525b	[hotfix] run cpu adam unittest in pytest (#424 )	2022-03-16 10:39:55 +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

Author

SHA1

Message

Date

LuGY

6a3f9fda83

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

2022-03-25 14:15:53 +08:00

Jiarui Fang

5d7dc3525b

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

2022-03-16 10:39:55 +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

3 Commits (4d322b79da72690d1cf46f88bf9d59f986e62040)