official/bert/model_training_utils.py · a009f4fb9d2fc4949e32192a944688925ef78659 · ModelZoo / ResNet50_tensorflow · GitLab

Find file Blame History Permalink

move collection trainable variables outside loop. · a009f4fb
Hongkun Yu authored Sep 03, 2019
```
add a flag to control loss scaling.

PiperOrigin-RevId: 267091566
```
a009f4fb

model_training_utils.py 16.5 KB