本文介绍了在Caffe中如何禁用特定层的反向传播，避免出现错误。提供了两种方法：一种是在prototxt文件中为层设置lrmult和decaymult参数；另一种是在loss层中设置propagate_down参数。
Original url:
https://blog.youkuaiyun.com/lanyuxuan100/article/details/78881501
在编写caffe某些层的时候，需要设置其不进行反向传播，否则会有类似如下的错误：
主要有两种方法解决：
（1）如果是C++写的层，直接在prototxt中层参数设置的时候，添加如下参数即可：
lr mult:0
decay mult:0
以faster-rcnn为例，其第一层与第二层卷积都使用了学习率为0.
（2）另外可以在loss 层的定义中增加如下参数即可：
propagate_down: 1
propagate_down: 0
扫一扫
1156
被折叠的 条评论
为什么被折叠?
到【灌水乐园】发言
