复制预训练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 } }