template <typename Dtype>
class SoftmaxWithLossLayer : public LossLayer<Dtype> {
public:
+ /**
+ * @param param provides LossParameter loss_param, with options:
+ * - ignore_label (optional)
+ * Specify a label value that should be ignored when computing the loss.
+ * - normalize (optional, default true)
+ * If true, the loss is normalized by the number of (nonignored) labels
+ * present; otherwise the loss is simply summed over spatial locations.
+ */
explicit SoftmaxWithLossLayer(const LayerParameter& param)
- : LossLayer<Dtype>(param),
- softmax_layer_(new SoftmaxLayer<Dtype>(param)) {}
+ : LossLayer<Dtype>(param) {}
virtual void LayerSetUp(const vector<Blob<Dtype>*>& bottom,
const vector<Blob<Dtype>*>& top);
virtual void Reshape(const vector<Blob<Dtype>*>& bottom,