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 } }
相關文章
相關標籤/搜索