微调caffemodel的一部分,并最后保存所有参数

时间:2017-11-19 04:00:29

标签: caffe

我想从现有的caffemodel微调模型。但是这个模型的一部分我不需要运行它,我想最后保存这部分参数,我该怎么办?

谢谢!

1 个答案:

答案 0 :(得分:1)

您可以为未微调的图层设置switch。只要它们存在于您的模型中,当您true微调网络到新的caffemodel时,它们的参数将与所有其他参数一起保存