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)
还没有任何评论哟~
