2015-09-22 38 views

回答

5

要訓練的網絡的權重與初始化/微調所用的權重之間需要有1-1對應關係。新老模型的架構必須匹配。

VGG-16的結構與models/finetune_flickr_style/train_val.prototxt(FlickrStyleCaffeNet)描述的結構不同。這是求解器將嘗試優化的網絡。即使它沒有崩潰,您加載的權重在新網絡中也沒有任何意義。

VGG-16網絡在Caffe模型動物園的this頁面上的deploy.prototxt文件中進行了描述。

+0

VGG是否有train_val.prototxt和solver.protxt?除了鏈接中描述的鏈接之外,我無法找到其中的鏈接,但人們認爲這不起作用。 – ytrewq

+1

還沒有碰到過VGG-16的train_val。看看[karpathy's](https://gist.github.com/ksimonyan/211839e770f7b538e2d8#gistcomment-1403727)train_val,看起來原型文件有點過時。你將以零權重結束訓練中不會改變的權重。您需要在整個網絡中添加初始化參數。一個簡單的選項就像weight_filler {type:「gaussian」std:0.01} bias_filler {type:「constant」value:0},但我建議查看VGG-16文章,瞭解作者使用的初始化。 – ypx

相關問題