Merge pull request davidsandberg#677 from Dobiasd/patch-1

replace keep_dims (deprecated in TF) with keepdims
stephenhuang8008 · Mar 31, 2018 · fc3b4c6 · fc3b4c6
2 parents fe78f66 + e13d5a9
commit fc3b4c6
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/align/detect_face.py b/src/align/detect_face.py
@@ -207,9 +207,9 @@ def fc(self, inp, num_out, name, relu=True):
     """
     @layer
     def softmax(self, target, axis, name=None):
-        max_axis = tf.reduce_max(target, axis, keep_dims=True)
+        max_axis = tf.reduce_max(target, axis, keepdims=True)
         target_exp = tf.exp(target-max_axis)
-        normalize = tf.reduce_sum(target_exp, axis, keep_dims=True)
+        normalize = tf.reduce_sum(target_exp, axis, keepdims=True)
         softmax = tf.div(target_exp, normalize, name)
         return softmax