experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Fix typo and use fully qualified names for consistency in LossScaleOptimizer docstring

Browse Source 
PiperOrigin-RevId: 214077127
This commit is contained in:
Smit Hinsu 2018-09-21 19:03:15 -07:00 committed by TensorFlower Gardener
parent 7229d08f0b
commit 812d5505f5
1 changed files with 3 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
tensorflow/contrib/mixed_precision/python/loss_scale_optimizer.py
View File

@ -66,10 +66,11 @@ class LossScaleOptimizer(optimizer.Optimizer):
# Choose a loss scale manager which decides how to pick the right loss scale
# throughout the training process.
loss_scale_manger = tf.contrib.mixed_precision.FixedLossScaleManager(5000)
loss_scale_manager = tf.contrib.mixed_precision.FixedLossScaleManager(5000)
# Wraps the original optimizer in a LossScaleOptimizer.
loss_scale_optimizer = LossScaleOptimizer(opt, loss_scale_manager)
loss_scale_optimizer =
tf.contrib.mixed_precision.LossScaleOptimizer(opt, loss_scale_manager)
# Call minimize() on the loss scale optimizer.
train_op = loss_scale_optimizer.minimize(loss)