caffe添加自己的层

首先修改src/caffe/proto/下的caffe.proto,修改好后需要编译

 

 

 

 

 然后修改include/caffe/layers/logwxl_layer.hpp

然后修改src/caffe/layers/logwxl_layer.cpp和logwxl_layer.cu

最后make all ,make test ,make runtest

在make runtest的时候,卧槽有一个错误,好像是没有updateV1parameter什么的,但是不影响最后的训练

posted @ 2016-12-19 16:22  simple_wxl  阅读(1616)  评论(0)    收藏  举报