c++ - 未调用 Caffe Layer 函数

Question

我已经在 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
...

多次调用损失层的原因(在上面的代码片段中)是，我使用自定义层中的层来调用其他层，(“Softmax” 类型的 softmax，sigmoid类型 "Sigmoid"，类型为 "Reshape" 的 reshape_softmax，reshape_sigmoid "Reshape"和另一个 "Reshape" 层。所有这 5 层都起作用在此自定义层的输入 blob 的不同部分上)

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

Answer 1

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