caffe SigmoidCrossEntropyLossLayer


代码:

layer {
  name: "loss_2"
  type: "SigmoidCrossEntropyLoss"
  bottom: "score_2"
  bottom: "label"
  top: "loss_2"
  loss_weight: 0.1
}
SigmoidCrossEntropyLossLayer的输入bottom[0],bottom[1],其中bottom[0]是输入的预测的结果,bottom[1]是标签值。bottom的维度都是(N×C×H×W)(N×C×H×W),bottom的表示符号是x,bottom[1]是p[0,1]

猜你喜欢

转载自blog.csdn.net/ziyouyi111/article/details/80884431