Commit Graph

4 Commits (3a7571b1d7b6904fc9f83bc7b53be18ad55aaf30)

Author SHA1 Message Date
LuGY 105c5301c3
[zero]added hybrid adam, removed loss scale in adam (#527)
3 years ago
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)
3 years ago