Commit Graph

7 Commits (c9ea317b367b61c3abc6496e76354fb297b4746c)

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