当我在Caffe上运行自己的图像时,程序在第0次迭代,测试网络(#0)处停止

我运行了Caffe并得到了以下输出:
enter image description here

谁能告诉我问题出在哪里?
我将不胜感激!!


回答:

看起来你的一个(或多个)标签值是无效的,请查看这个PR获取更多信息:

如果你有一个无效的真实标签,"SoftmaxWithLoss"会悄悄地访问无效的内存 […] 旧的检查仅在DEBUG模式下有效,并且也只对CPU有效。

确保你的预测向量长度与你试图预测的标签数量相匹配。

从你的评论来看,似乎你的标签范围在0..10575之间,但另一方面,你的分类层"fc7"仅预测1000个类别的概率。因此,"SoftmaxWithLoss"层试图计算预测标签l>1000的损失,并访问概率数组之外的内存,导致了段错误。

Related Posts

L1-L2正则化的不同系数

我想对网络的权重同时应用L1和L2正则化。然而,我找不…

使用scikit-learn的无监督方法将列表分类成不同组别,有没有办法?

我有一系列实例,每个实例都有一份列表,代表它所遵循的不…

f1_score metric in lightgbm

我想使用自定义指标f1_score来训练一个lgb模型…

通过相关系数矩阵进行特征选择

我在测试不同的算法时,如逻辑回归、高斯朴素贝叶斯、随机…

可以将机器学习库用于流式输入和输出吗?

已关闭。此问题需要更加聚焦。目前不接受回答。 想要改进…

在TensorFlow中,queue.dequeue_up_to()方法的用途是什么?

我对这个方法感到非常困惑,特别是当我发现这个令人费解的…

发表回复

您的邮箱地址不会被公开。 必填项已用 * 标注