Advertisement

caffe net layer

阅读量:

2016年4月,即Caffe进行重大版本更新之后(每个层都对应一个同名的cpp和hpp文件)

描述一下本次要实现层的功能:

正向传播过程中直接复制数据,反向传播时则按照设定比例对梯度进行缩放。

该结构在某些特定的网络架构中具有重要的支持功能,例如当网络包含多个分支时,若希望某一特定分支不对前面层的参数更新产生影响,可将该分支对应的梯度缩放比例设置为0。

(1)创建HPP头文件diff_cutoff_layer.hpp

不同功能类型的层所对应的头文件存在差异,具体可前往“caffe/include/caffe/layers”目录中查找相似的现成文件作为参考。本次编写的hpp文件最终也需放置于该目录中。

复制代码
    //*****************************************
    #ifndef CAFFE_DIFFCUTOFF_LAYER_HPP_
    #define CAFFE_DIFFCUTOFF_LAYER_HPP_
    //*****************************************
    
    #include <vector>
    #include "caffe/blob.hpp"
    #include "caffe/layer.hpp"
    #i

全部评论 (0)

还没有任何评论哟~