Commit Graph

8 Commits (cc311bdf95eeb8dc7f77b51ecf030d51335abcaa)

Author SHA1 Message Date
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