作者:好人langren_840 | 来源:互联网 | 2023-08-18 10:28
Originalurl:https:blog.csdn.netlanyuxuan100articledetails78881501在编写caffe某些层的时候,需要设置
Original url:
https://blog.csdn.net/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