Caffe SoftmaxWithLoss错误

作者: 冷月如霜·胡狼
发布时间: 2024-12-25 09:48:06 (2天前)
转自：

2 条回复

0#
回复此人
v-star*위위 | 2019-08-31 10-32

<div class =“post-text”itemprop =“text”> <P> 使用时 <code> "SoftmaxWithLoss" </code> 用于预测二进制标签的图层，您的“类概率”向量应为长度为2（而不是1）。 <BR/> 你收到一个错误，说你的“类概率”向量（又名“ <code> prob_ </code> “）的尺寸为1，而它应至少为2（即的<strong> 严格 </强> 大于最大标签）。 </p> <P> 校验 <code> num_output </code> 生成类概率的图层中的参数。 </p> <P> 或者，对于二进制分类，请考虑使用 <a href="http://caffe.help/manual/layers/sigmoidcrossentropyloss.html" rel="nofollow noreferrer"> <code> "SigmoidCrossEntropyLoss" </code> </A> </p> </DIV>

编辑

登录后才能参与评论