LossScaleOptimizer¶

An optimizer that dynamically scales the loss to prevent underflow.

Abstract Signature:

LossScaleOptimizer(inner_optimizer, initial_scale: float = 32768.0, dynamic_growth_steps: int = 2000)

API: keras.optimizers.LossScaleOptimizer

Strategy: Direct Mapping

API: tf.keras.mixed_precision.LossScaleOptimizer

Strategy: Direct Mapping