diff --git a/tensorflow/python/keras/mixed_precision/experimental/loss_scale.py b/tensorflow/python/keras/mixed_precision/experimental/loss_scale.py index d8e173a07f6..174ab4d5b86 100644 --- a/tensorflow/python/keras/mixed_precision/experimental/loss_scale.py +++ b/tensorflow/python/keras/mixed_precision/experimental/loss_scale.py @@ -80,7 +80,7 @@ class LossScale(trackable.Trackable): grads: A list of unscaled gradients, each which is the gradient of the loss with respect to a weight. The gradients should have already been divided by the loss scale being before passed to this function. 'None' - gradients are accepted, and should be ignored. + gradients are accepted, and are ignored. Returns: update_op: In eager mode, None. In graph mode, an op to update the loss diff --git a/tensorflow/python/training/experimental/loss_scale.py b/tensorflow/python/training/experimental/loss_scale.py index a67823347bb..47f0bc67a03 100644 --- a/tensorflow/python/training/experimental/loss_scale.py +++ b/tensorflow/python/training/experimental/loss_scale.py @@ -84,7 +84,7 @@ class LossScale(trackable.Trackable): grads: A list of unscaled gradients, each which is the gradient of the loss with respect to a weight. The gradients should have already been divided by the loss scale being before passed to this function. 'None' - gradients are accepted, and should be ignored. + gradients are accepted, and are ignored. Returns: update_op: In eager mode, None. In graph mode, an op to update the loss