Fix some deprecation warning in losses.py.

google · Aug 20, 2019 · f6620e7 · f6620e7
1 parent f51dc7d
commit f6620e7
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/losses.py b/losses.py
@@ -47,8 +47,8 @@ def calculate_loss(self,
     with tf.name_scope("loss_xent"):
       epsilon = 1e-5
       float_labels = tf.cast(labels, tf.float32)
-      cross_entropy_loss = float_labels * tf.log(predictions + epsilon) + (
-          1 - float_labels) * tf.log(1 - predictions + epsilon)
+      cross_entropy_loss = float_labels * tf.math.log(predictions + epsilon) + (
+          1 - float_labels) * tf.math.log(1 - predictions + epsilon)
       cross_entropy_loss = tf.negative(cross_entropy_loss)
       if label_weights is not None:
         cross_entropy_loss *= label_weights