Caffe编译中的in expansion of macro ‘STUB_GPU_BACKWARD’类型错误解决方案

[  1%] Building CXX object src/caffe/CMakeFiles/caffe.dir/layers/sigmoid_cross_entropy_loss_layer.cpp.o
In file included from /home/lzh/Caffe/hed-master/include/caffe/common.hpp:19:0,
                 from /home/lzh/Caffe/hed-master/include/caffe/blob.hpp:8,
                 from /home/lzh/Caffe/hed-master/include/caffe/layer.hpp:9,
                 from /home/lzh/Caffe/hed-master/src/caffe/layers/sigmoid_cross_entropy_loss_layer.cpp:5:
/home/lzh/Caffe/hed-master/include/caffe/util/device_alternate.hpp:30:39: error: no ‘void caffe::SigmoidCrossEntropyLossLayer<Dtype>::Backward_gpu(const std::vector<caffe::Blob<Dtype>*>&, const std::vector<bool>&, const std::vector<caffe::Blob<Dtype>*>&)’ member function declared in class ‘caffe::SigmoidCrossEntropyLossLayer<Dtype>const vector<Blob<Dtype>*>& bottom) { NO_GPU; } \
                                       ^
/home/lzh/Caffe/hed-master/src/caffe/layers/sigmoid_cross_entropy_loss_layer.cpp:121:1: note: in expansion of macro ‘STUB_GPU_BACKWARD’
 STUB_GPU_BACKWARD(SigmoidCrossEntropyLossLayer, Backward);
 ^
src/caffe/CMakeFiles/caffe.dir/build.make:566: recipe for target 'src/caffe/CMakeFiles/caffe.dir/layers/sigmoid_cross_entropy_loss_layer.cpp.o' failed
make[2]: *** [src/caffe/CMakeFiles/caffe.dir/layers/sigmoid_cross_entropy_loss_layer.cpp.o] Error 1
CMakeFiles/Makefile2:272: recipe for target 'src/caffe/CMakeFiles/caffe.dir/all' failed
make[1]: *** [src/caffe/CMakeFiles/caffe.dir/all] Error 2
Makefile:127: recipe for target 'all' failed
make: *** [all] Error 2

1.解决方案

这里以上面的错误为例,其他类似的错误解决方法大同小异!

首先打开include/caffe/loss_layers.hpp,然后找到SigmoidCrossEntropyLossLayer类,取消下面的注释即可

  //virtual void Backward_gpu(const vector<Blob<Dtype>*>& top,
      //const vector<bool>& propagate_down, const vector<Blob<Dtype>*>& bottom);

2.错误分析

同样还是以这个错误为例!

其实下面这句话已经说的很明白了,意思就是在SigmoidCrossEntropyLossLayer类里面没有Backward_gpu()这个函数

device_alternate.hpp:30:39: error: no ‘void caffe::SigmoidCrossEntropyLossLayer<Dtype>::Backward_gpu(const std::vector<caffe::Blob<Dtype>*>&, const std::vector<bool>&, const std::vector<caffe::Blob<Dtype>*>&)’ member function declared in class 

所以我们只要在定义SigmoidCrossEntropyLossLayer类的地方加上这个函数即可,一般在类的定义处都会有被注释的这个函数,若是没有,手动加上即可.

需要注意一点的是,loss类的layer定义在loss_layers.hpp里;卷积类的layer定义在vision_layers.hpp里;data类的layer定义在data_layers.hpp里

最后这是针对旧版caffe的,若是新版的caffe(每一个layer都有对应的hpp和cpp),大家自行变通一下,反正原理都差不多的

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值