Caffe:检查失败:网络应该只有一个输出

时间:2016-01-06 16:08:55

标签: android machine-learning neural-network deep-learning caffe

我正在使用Caffe android lib(基于C ++ API)。我正在尝试使用GoogleNet模型 不幸的是,它给了我这个错误:

order_id

我有一种感觉,我错过了一些愚蠢的东西。有什么想法吗?

1 个答案:

答案 0 :(得分:0)

您使用的是GoogLeNet的哪些实现?
根据实现情况,您可能确实获得了多个输出。 最初的GoogLeNet(在CVPR 2015中发布)有三个"输出"训练损失,因此您可能正在使用由多个输出层产生的类似原型。

考虑this deploy.prototxt - 它只有一个输出"prob"。但是,相应的train_val.prototxt有三个输出:"loss1/loss1""loss2/loss1""loss3/loss3"