我正在复制步骤
http://caffe.berkeleyvision.org/gathered/examples/finetune_flickr_style.html
我想将网络更改为在以下位置获得的VGG模型
http://www.robots.ox.ac.uk/~vgg/software/very_deep/caffe/VGG_ILSVRC_16_layers.caffemodel
只需按以下方式替换模型参数就足够了吗?
./build/tools/caffe train -solver models/finetune_flickr_style/solver.prototxt -weights VGG_ISLVRC_16_layers.caffemodel -gpu 0
还是我需要调整学习率,迭代次数,即它是否带有单独的prototxt文件?
最佳答案
要训练的网络的权重和用于初始化/微调的权重之间必须存在1-1对应关系。新旧模型的架构必须匹配。
VGG-16具有与models / finetune_flickr_style / train_val.prototxt(FlickrStyleCaffeNet)描述的模型不同的体系结构。这是求解器将尝试优化的网络。即使它没有崩溃,您已加载的权重在新网络中也没有任何意义。
Caffe的Model Zoo中this页上的deploy.prototxt文件中描述了VGG-16网络。