Skip to content

Commit

Permalink
[keras/mixed_precision/loss_scale_optimizer.py] Docstring minor impro…
Browse files Browse the repository at this point in the history
…vements
  • Loading branch information
SamuelMarks committed Apr 23, 2023
1 parent b43b215 commit 19dcdeb
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion keras/mixed_precision/loss_scale_optimizer.py
Original file line number Diff line number Diff line change
Expand Up @@ -406,7 +406,7 @@ class BaseLossScaleOptimizer(metaclass=LossScaleOptimizerMetaclass):
Args:
inner_optimizer: The `tf.keras.optimizers.Optimizer` or
`tf.keras.optimizers.experimental.Optimizer` instance to wrap.
dynamic: Bool indicating whether dynamic loss scaling is used. If True,
dynamic: Bool indicating whether dynamic loss scaling is used. If `True`,
the loss scale will be dynamically updated over time using an algorithm
that keeps the loss scale at approximately its optimal value. If False,
a single fixed loss scale is used and `initial_scale` must be
Expand Down

0 comments on commit 19dcdeb

Please sign in to comment.