caffe fine tune 复制预训练model的参数和freeze指定层参数

复制预训练model的参数,只须要从新copy一个train_val.prototxt。而后把不须要复制的层的名字改一下,如(fc7 -> fc7_new),而后fine tune便可。学习

freeze指定层参数,只须要把对应层的学习率lr_mult 设置为0便可,如:it

在layer里面加上param { lr_mult: 0 }就能够了,好比全链接层里面: layer { type: "InnerProduct" param { # 对应第1个参数blob的配置,也就是全链接层的参数矩阵的配置 lr_mult: 0 # 学习率为0,其余参数能够看caffe.proto里面的ParamSpec这个类型 } param { # 对应第2个参数blob的配置,也就是全链接层的偏置项的配置 lr_mult: 0 # 学习率为0 } }
相关文章
相关标签/搜索