Update docstring assertiveness
This commit is contained in:
parent
3b486c4464
commit
395c93bbf9
@ -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
|
||||
|
@ -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
|
||||
|
Loading…
x
Reference in New Issue
Block a user