| title | Softmax with Loss Layer |
|---|
- Layer type:
SoftmaxWithLoss - Doxygen Documentation
- Header:
./include/caffe/layers/softmax_loss_layer.hpp - CPU implementation:
./src/caffe/layers/softmax_loss_layer.cpp - CUDA GPU implementation:
./src/caffe/layers/softmax_loss_layer.cu
The softmax loss layer computes the multinomial logistic loss of the softmax of its inputs. It's conceptually identical to a softmax layer followed by a multinomial logistic loss layer, but provides a more numerically stable gradient.
- Parameters (
SoftmaxParameter softmax_param) - From
./src/caffe/proto/caffe.proto:
{% highlight Protobuf %} {% include proto/SoftmaxParameter.txt %} {% endhighlight %}
- Parameters (
LossParameter loss_param) - From
./src/caffe/proto/caffe.proto:
{% highlight Protobuf %} {% include proto/LossParameter.txt %} {% endhighlight %}