тонкой настройки с VGG на CAFFE

Question

Я тиражирование шаги в 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 файлами?

Answer 1

Должно быть соответствие 1-1 между весами сети, которую вы хотите обучить, и весами, которые вы используете для инициализации/точной настройки. Архитектура старой и новой модели должна соответствовать.

VGG-16 отличается своей архитектурой, чем модель, описанная в моделях/finetune_flickr_style/train_val.prototxt (FlickrStyleCaffeNet). Это сеть, которую решатель попытается оптимизировать. Даже если он не падает, загруженные вами веса не имеют никакого значения в новой сети.

Сеть VGG-16 описана в файле deploy.prototxt на странице this в Cooe's Model Zoo.