Merge pull request #83 from sfujiwara/bug-fix-kaffe

ethereon · web-flow · commit d870c51e8fa3 · 2017-01-16T16:17:58.000+05:30
bug fix for softmax layer in network.py
diff --git a/kaffe/tensorflow/network.py b/kaffe/tensorflow/network.py
@@ -212,7 +212,7 @@ def softmax(self, input, name):
                 input = tf.squeeze(input, squeeze_dims=[1, 2])
             else:
                 raise ValueError('Rank 2 tensor input expected for softmax!')
-        return tf.nn.softmax(input, name)
+        return tf.nn.softmax(input, name=name)
 
     @layer
     def batch_normalization(self, input, name, scale_offset=True, relu=False):
