caffe Python API 之SoftmaxWithLoss

net.loss = caffe.layers.SoftmaxWithLoss(net.fc3, net.label)
输出:
layer {
  name: "loss"
  type: "SoftmaxWithLoss"
  bottom: "fc3"
  bottom: "label"
  top: "loss"
}

猜你喜欢

转载自www.cnblogs.com/houjun/p/9912457.html
今日推荐