第一种方法:将原来的prototxt中所有的fc8改为fc8-re。(若希望修改层的学习速度比其他层更快一点,可以将lr_mult改为原来的10倍或者其他倍数)第二种方法:只修改name,如下例子所示: layer { name: "fc8-re" #原来为"fc8" type: "InnerProduct" bottom: "fc7" top: "fc8" param { lr_mult: 1.0 decay_mult: 1.0 } param { lr_mult: 2.0 decay_mult: 0.0 } inner_product_param { num_output: 5 #原来为"1000" weight_filler { type: "gaussian" std: 0.01 } bias_filler { type: "constant" value: 0.0 } } }
Caffe模型微调技巧
本文介绍两种在Caffe框架中进行模型微调的方法。第一种是全局替换原模型的fc8层为fc8-re,并调整学习速率。第二种是仅修改层名,详细展示了如何在配置文件中进行具体参数设置,包括输出数量、填充方式及学习率等。
794





