[FIX](loss) Add default reduction in SigmoidCrossEntropyLoss

wbenbihi · Aug 25, 2022 · c3460a5 · c3460a5
1 parent 6c05364
commit c3460a5
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/hourglass_tensorflow/losses/sigmoid_cross_entropy.py b/hourglass_tensorflow/losses/sigmoid_cross_entropy.py
@@ -3,7 +3,9 @@
 
 
 class SigmoidCrossEntropyLoss(keras.losses.Loss):
-    def __init__(self, reduction=..., name=None, *args, **kwargs):
+    def __init__(
+        self, reduction=tf.keras.losses.Reduction.AUTO, name=None, *args, **kwargs
+    ):
         super().__init__(reduction, name)
 
     def call(self, y_true, y_pred):