版权声明:如需转载请评论告知并标明出处 https://blog.csdn.net/ShuqiaoS/article/details/83269687
自定义Caffe层遇到的问题
博主在自定义Caffe层的时候遇到了一些问题,在此记录下来并附上解决方案。更新于2018.10.22。
1
在编译时报如下错误:
src/caffe/layers/embed_layer2d.cu(57): error: function "caffe::MyLayer<Dtype>::Backward_gpu [with Dtype=float]" cannot be instantiated -- no template definition was supplied
src/caffe/layers/embed_layer2d.cu(57): error: function "caffe::MyLayer<Dtype>::Backward_gpu [with Dtype=double]" cannot be instantiated -- no template definition was supplied
原因:在hpp文件中没有注释掉或修改关于backward_gpu的定义。修改可以参照:
virtual void Backward_gpu(const vector<Blob<Dtype>*>& top,
const vector<bool>& propagate_down, const vector<Blob<Dtype>*>& bottom) { for(int i=0; i<propagate_down.size(); i++) if(propagate_down[i]) LOG(FATAL) << "Embed2DLayer cannot do backward."; return; }
2
报错地址必须是整型:
error: expression must have integral or unscoped enum type
在cu文件编写的时候要注意,bottom_data[index]和top_data[index]后面定义地址的参数index的类型。
3
error: expression must be a modifiable lvalue
这是由于等号左边的值无法修改。博主遇到这个报错的原因是,左侧的地址并非当前的index。博主对于这个的理解是:Caffe无法直接跨线程赋值!!