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


猜你喜欢

转载自blog.csdn.net/junmuzi/article/details/79705758