老师，您好

来源：6-5 VGG16预训练模型读取函数封装

慕粉1471439700

2018-10-24

请问如果训练好的模型是2分类的，我想改变改变全连接的输出为10分类的，但是重用之前训练好的模型参数，请问有什么办法吗？

写回答

1回答

正十七

2018-10-30

我们在调参的课程中讲过fine-tune，这个技术符合你的需求。具体来说就是：

保持前面的网络结构然后把除了最后一层的参数载入进来做初始化。最后一层的参数随机初始化，然后继续训练就可以了。

深度学习之神经网络(CNN/RNN/GAN)算法原理+实战

深度学习算法工程师必学，深入理解深度学习核心算法CNN RNN GAN

2617 学习 · 935 问题

相似问题

回答 1

回答 7

回答 1

回答 1

回答 1