SE-Resnet踩过的坑

最近项目没什么事情，导师又让写论文，所以开始跑实验，争取研一下学期发篇小论文吧。 在跑SE-Resnet的时候，会要自己在caffe的源码中添加一个新的层，叫Axpy，将avg pooling及scale等操作一起封装的一个type. 从github上下载了源码之后： 添加的过程很简单，将.hpp,.cpp,.cu文件分别放入到对应的文件夹，在caffe.proto文件中加上想对应的message

>>阅读原文<<