首页 > 代码库 > caffe添加自己的层
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什么的,但是不影响最后的训练
caffe添加自己的层
声明:以上内容来自用户投稿及互联网公开渠道收集整理发布,本网站不拥有所有权,未作人工编辑处理,也不承担相关法律责任,若内容有误或涉及侵权可进行投诉: 投诉/举报 工作人员会在5个工作日内联系你,一经查实,本站将立刻删除涉嫌侵权内容。