official/nlp/optimization.py · 4f50e2fc6b4d534a0cb5aa33789fe04edfba5c52 · ModelZoo / ResNet50_tensorflow

Ran Chen authored Oct 22, 2020

This fix tensorflow_models/official/nlp/bert/run_pretraining failure with
explicit allreduce.

PiperOrigin-RevId: 338521478

4f50e2fc

optimization.py 9.44 KB