caffe中如何設置某層不參與反向傳播

在編寫caffe某些層的時候,需要設置其不進行反向傳播,否則會有類似如下的錯誤: 主要有兩種方法解決: (1)如果是C++寫的層,直接在prototxt中層參數設置的時候,添加如下參數即可: lr mult:0 decay mult:0 以faster-rcnn爲例,其第一層與第二層卷積都使用了學習率爲0. (2)另外可以在loss 層的定義中增加如下參數即可:     propagate_dow
相關文章
相關標籤/搜索