未调用 Caffe Layer 函数答案

【问题标题】：Caffe Layer function is not invoked未调用 Caffe Layer 函数
【发布时间】：2017-12-20 05:38:01
【问题描述】：

我在 caffe (C++) 中编写了一个自定义层。在运行此代码时（训练使用该层的模型），设置层（层的"LayerSetUp" 方法）被调用（通过编写一段代码来验证，该代码将在硬盘上创建一个文件并转储一些数据）。但是在执行期间似乎没有调用Forward_cpu() 和Backward_cpu() 方法。可能的原因是什么？

这是运行模型以训练使用此自定义损失层的输出。

...
I0715 09:23:57.415463 31256 net.cpp:84] Creating Layer loss
I0715 09:23:57.415472 31256 net.cpp:406] loss <- permute_conv11
I0715 09:23:57.415482 31256 net.cpp:406] loss <- bbox
I0715 09:23:57.415495 31256 net.cpp:380] loss -> loss
I0715 09:23:57.433014 31256 layer_factory.hpp:77] Creating layer loss
I0715 09:23:57.437386 31256 layer_factory.hpp:77] Creating layer loss
I0715 09:23:57.438171 31256 layer_factory.hpp:77] Creating layer loss
I0715 09:23:57.438897 31256 layer_factory.hpp:77] Creating layer loss
I0715 09:23:57.438989 31256 layer_factory.hpp:77] Creating layer loss
I0715 09:23:57.440030 31256 net.cpp:122] Setting up loss
I0715 09:23:57.440052 31256 net.cpp:129] Top shape: (1)
I0715 09:23:57.440058 31256 net.cpp:132]     with loss weight 1
I0715 09:23:57.440099 31256 net.cpp:137] Memory required for data: 3146726596
...

损失层被多次调用（在上面的sn-p中）的原因是，我使用自定义层中的层来调用其他层，（"Softmax"类型的softmax，"Sigmoid"类型的sigmoid， "Reshape" 类型的 reshape_softmax、reshape_sigmoid "Reshape" 和另一个 "Reshape" 层。所有这 5 层都作用于此自定义层的输入 blob 的不同部分）

在训练模型时似乎根本没有调用 Forward_cpu() 方法。可能是什么问题，我该如何解决？

【问题讨论】：

标签： c++ neural-network caffe layer

【解决方案1】：

在这种情况下很可能调用了Forward_gpu() 方法。要克服此错误，请确保您没有实现 Forward_gpu 和 Backward_gpu() 的 <your_custom_layer>.cu 文件

【讨论】：

是的，这就是问题所在！非常感谢