Commit Graph

4 Commits (0ed7042f424ae9b62920ff5a85b41c57e9d51e0d)

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