SE-Resnet踩過的坑

最近項目沒什麼事情，導師又讓寫論文，所以開始跑實驗，爭取研一下學期發篇小論文吧。 在跑SE-Resnet的時候，會要自己在caffe的源碼中添加一個新的層，叫Axpy，將avg pooling及scale等操作一起封裝的一個type. 從github上下載了源碼之後： 添加的過程很簡單，將.hpp,.cpp,.cu文件分別放入到對應的文件夾，在caffe.proto文件中加上想對應的message

>>阅读原文<<