diff --git a/tensorflow_addons/losses/focal_loss.py b/tensorflow_addons/losses/focal_loss.py index b7f9c946ad..b0c1a59876 100644 --- a/tensorflow_addons/losses/focal_loss.py +++ b/tensorflow_addons/losses/focal_loss.py @@ -68,7 +68,7 @@ def __init__( from_logits: bool = False, alpha: FloatTensorLike = 0.25, gamma: FloatTensorLike = 2.0, - reduction: str = tf.keras.losses.Reduction.NONE, + reduction: str = tf.keras.losses.Reduction.AUTO, name: str = "sigmoid_focal_crossentropy", ): super().__init__(