Commit Graph

18 Commits (b38e4f2840ebc4ee0195ab8de789bf5b8d54ef37)

Author SHA1 Message Date
Zhang Ting 906e7f921e
add fuse_bn_act op (#27230)
4 years ago
Zhen Wang d708b21074
Update amp_check_finite_and_scale_op and add an updating_loss_scaling op for static graph amp training. (#26240)
5 years ago
Zhen Wang bcdbac1753
fix some cast error. (#26884)
5 years ago
mapingshuo f0e743f136
fix AMP and recompute (#23551)
5 years ago
gongweibao 5e07db15e6
fix init scaling value test=develop (#22145)
5 years ago
Zhen Wang be2e3e67d9
Fix some typos in AMP. (#21354)
5 years ago
gongweibao 3255fe69bb Add custom black variable name set in amp interface. (#20875)
5 years ago
gongweibao 1d82025e89
Add interface so user can get scaled loss when they use customized loss. (#20571)
5 years ago
gongweibao 7b9e33972a
delete backward return list test=develop (#20294)
5 years ago
Jie Fang d9db94d752 Optimize amp for multi-gpu to enable FP16 gradients transfer across gpus. (#19714)
6 years ago
gongweibao 6c2bc29cc0
Fix float16 optimizer. (#19682)
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
Zeng Jinle 5dce1da680 remove reset recordio usage (#19519)
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)
6 years ago
Yibing Liu beda78258f
Init mixed precision training interface (#16856)
6 years ago