【深度学习框架Caffe学习与应用】第五课 自定义神经层和数据输入层

一、自定义神经层 1.创建新定义的头文件,目录在caffe/include/caffe/layers/my_neuron_layer.hpp   我复制了conv_layer.hpp文件,然后在该文件的内容上进行更改 如果只是需要CPU方法,可以注释掉forward_gpu和backward_gpu 修改对照内容如下(左侧是conv_layer.hpp,右侧是my_neuron_layer.hpp
相关文章
相关标签/搜索