Commit Graph

13 Commits (cf475f95dfa1c28362b4fdb94930824054c3dce4)

Author SHA1 Message Date
Zhen Wang be2e3e67d9
Fix some typos in AMP. (#21354)
7 years ago
gongweibao 3255fe69bb Add custom black variable name set in amp interface. (#20875)
7 years ago
gongweibao 1d82025e89
Add interface so user can get scaled loss when they use customized loss. (#20571)
7 years ago
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