LSoftmaxTensorFlow

November 21, 2017 ยท View on GitHub

TensorFlow version of L_SoftMax.

Results:

results

I found prelu is quite stable than relu, so I used prelu as paper said.

Reference:

  • mx-lsoftmax
  • Large-Margin Softmax Loss for Convolutional Neural Networks

Contribution

This is mainly implemented by py_func, which is quite slow. If anyone have implemented a tf_op in C++ or cuda, pull request is warmly welcome.

Contents

  1. 0.1Results:
  2. 0.2Reference:
  3. 0.3Contribution