Commit Graph

10 Commits (7f1ad510bd5df6093767ab65817bd2f9cd970a7f)

Author SHA1 Message Date
tianshuo78520a 433cef03e5
fix typo word (#22784)
6 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
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
Tao Luo 2f8c7e021f
remove unused inference_transpiler unit-tests (#19130)
7 years ago
Yibing Liu f9796b1249
Add LAMB Optimizer support (#17489)
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