fix categorical_crossentropy normalizer

hellosummer8850 · Jun 15, 2020 · 06a8a74 · 06a8a74
1 parent 71945ab
commit 06a8a74
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/efficientdet/det_model_fn.py b/efficientdet/det_model_fn.py
@@ -165,7 +165,7 @@ def focal_loss(y_pred, y_true, alpha, gamma, label_smoothing, normalizer):
 
     # Get the cross_entropy for each entry
     ce = tf.keras.losses.categorical_crossentropy(
-        y_true, y_pred, from_logints=True, label_smoothing=label_smoothing)
+        y_true, y_pred, from_logits=True, label_smoothing=label_smoothing)
 
     pred_prob = tf.sigmoid(y_pred)