gpt4 book ai didi

neural-network - 在caffe上使用VGG进行微调

转载 作者:行者123 更新时间:2023-12-04 04:31:37 25 4
gpt4 key购买 nike

我正在复制以下步骤 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) 描述的模型不同。这是求解器将尝试优化的网络。即使它没有崩溃,您加载的权重在新网络中也没有任何意义。

VGG-16 网络在this 上的deploy.prototxt 文件中进行了描述。 Caffe 模型动物园中的页面。

关于neural-network - 在caffe上使用VGG进行微调,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/32717054/

25 4 0
Copyright 2021 - 2024 cfsdn All Rights Reserved 蜀ICP备2022000587号
广告合作:1813099741@qq.com 6ren.com