Changing the deprecated parameter name dim to axis

sadeepj · Mar 3, 2018 · 64a459f · 64a459f
1 parent 8c15674
commit 64a459f
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/crfrnn_layer.py b/src/crfrnn_layer.py
@@ -88,7 +88,7 @@ def call(self, inputs):
         q_values = unaries
 
         for i in range(self.num_iterations):
-            softmax_out = tf.nn.softmax(q_values, dim=0)
+            softmax_out = tf.nn.softmax(q_values, axis=0)
 
             # Spatial filtering
             spatial_out = custom_module.high_dim_filter(softmax_out, rgb, bilateral=False,