首先修改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什么的,但是不影响最后的训练
本站文章如无特殊说明,均为本站原创,如若转载,请注明出处:caffe添加自己的层 - Python技术站