Caffe | 自定义参数和结构
发布时间
阅读量：
阅读量
1.自定义字段设计与实现
在使用较早版本的caffe框架运行resnext网络过程中，遇到了如下的错误问题。正如前一篇文章Caffe | 核心积木Layer层类详解中所提及的，在caffe.proto文件内的PoolingParameter配置项中缺失了ceil_mode这一字段。为此，必须在源代码中补充该参数及其对应的实现逻辑，并完成caffe的重新编译操作。
Message type “caffe.PoolingParameter” has no field named “ceil_mode”.
（1）对pooling_layer.hpp文件进行修改
首先在头文件中引入参数的声明，由于在C/C++语言中，参数必须在使用前完成定义。
int height_, width_;
int pooled_height_, pooled_width_;
bool global_pooling_;
bool ceil_mode_; //添加bug中指出的缺少的属性
Blob<Dtype> rand_idx_;
Blob<int> max_idx_;全部评论 (0)
还没有任何评论哟~