2

http://caffe.berkeleyvision.org/gathered/examples/finetune_flickr_style.htmlの手順を複製しています

ネットワークをhttp://www.robots.ox.ac.uk/~vgg/software/very_deep/caffe/VGG_ILSVRC_16_layers.caffemodelで取得した VGG モデルに変更したい

モデルパラメータを次のように置き換えるだけで十分ですか?

./build/tools/caffe train -solver models/finetune_flickr_style/solver.prototxt -weights VGG_ISLVRC_16_layers.caffemodel -gpu 0

それとも、学習率や反復を調整する必要がありますか? つまり、別の prototxt ファイルが付属していますか?

4

1 に答える 1

5

トレーニングするネットワークの重みと、初期化/微調整に使用する重みの間には、1 対 1 の対応が必要です。古いモデルと新しいモデルのアーキテクチャは一致している必要があります。

VGG-16 は、models/finetune_flickr_style/train_val.prototxt (FlickrStyleCaffeNet) で説明されているモデルとは異なるアーキテクチャを持っています。これは、ソルバーが最適化しようとするネットワークです。クラッシュしなくても、ロードした重み​​は新しいネットワークでは何の意味もありません。

VGG-16 ネットワークは、Caffe の Model Zoo のこのページにある deploy.prototxt ファイルに記述されています。

于 2015-09-22T15:10:00.303 に答える