Commit Graph

10 Commits (e4033a06d55429acbb97ffd887d6268e83351986)

Author SHA1 Message Date
mapingshuo f0e743f136
fix AMP and recompute (#23551)
6 years ago
Zhen Wang be2e3e67d9
Fix some typos in AMP. (#21354)
6 years ago
gongweibao 3255fe69bb Add custom black variable name set in amp interface. (#20875)
6 years ago
Jie Fang d9db94d752 Optimize amp for multi-gpu to enable FP16 gradients transfer across gpus. (#19714)
6 years ago
Jie Fang c6a598a276 init new amp, optimize inserting cast op for batchnorm (#18596)
6 years ago
gongweibao abaf87be2b
Change backward_guard to optimize_guard to maximize the allreduce overlap. (#19506)
6 years ago
Jie Fang 2b4ef509ea init custom black white list (#18377)
6 years ago
Jie Fang 172c2facef init black/white lists (#17847)
6 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