Commit Graph

10 Commits (12e4be038262fcb33b089710edaed2fe62320e3d)

Author SHA1 Message Date
gongweibao 7b9e33972a
delete backward return list test=develop (#20294)
7 years ago
Jie Fang d9db94d752 Optimize amp for multi-gpu to enable FP16 gradients transfer across gpus. (#19714)
7 years ago
gongweibao 6c2bc29cc0
Fix float16 optimizer. (#19682)
7 years ago
Jie Fang c6a598a276 init new amp, optimize inserting cast op for batchnorm (#18596)
7 years ago
gongweibao abaf87be2b
Change backward_guard to optimize_guard to maximize the allreduce overlap. (#19506)
7 years ago
Zeng Jinle 5dce1da680 remove reset recordio usage (#19519)
7 years ago
Jie Fang 2b4ef509ea init custom black white list (#18377)
7 years ago
Jie Fang 172c2facef init black/white lists (#17847)
7 years ago
Jie Fang 30e178fa2c init auto loss scaling (#17194)
7 years ago
Yibing Liu beda78258f
Init mixed precision training interface (#16856)
7 years ago