我已经使用caffe
训练了一个回归网络。我在列车和测试阶段使用"EuclideanLoss"
层。我已经绘制了这些,结果看起来很有希望。
现在我想部署模型并使用它。我知道如果使用SoftmaxLoss
,则部署文件中的最后一层必须为Softmax
。在Euclidean loss
的情况下,这应该是什么?
答案 0 :(得分:2)
对于部署,您只需要丢弃损失层,在您的情况下是"EuclideanLoss"
图层。您的网络输出是您输入损失图层的"bottom"
。
对于"SoftmaxWithLoss"
图层(和"SigmoidCrossEntropy"
),您需要替换损失图层,因为损失图层中包含一个额外的图层(出于计算原因)。 / p>