当我在caffe上运行我自己的图像时,它会在Iteration 0,测试网(#0)处停止

时间:2016-03-23 09:00:09

标签: machine-learning segmentation-fault neural-network deep-learning caffe

我跑了caffe并得到了这个输出:
enter image description here

谁能告诉我这是什么问题?
我真的很感激!!

1 个答案:

答案 0 :(得分:0)

您的标签值中的一个(或多个)似乎无效,有关信息,请参阅this PR

  

如果你有一个无效的地面实况标签,"SoftmaxWithLoss"将默默地访问无效的内存[...]旧的检查仅在DEBUG模式下工作,也只适用于CPU。

确保您的预测矢量长度与您尝试预测的标签数量相匹配。

根据您的评论,您似乎有0..10575范围内的标签,但另一方面,您的分类图层"fc7"仅预测1000个类的概率。因此,"SoftmaxWithLoss"图层会尝试计算用于预测标签l> 1000的损失,并在概率数组外部访问内存,从而导致分段错误。