问题描述
有一个可以训练对象的预训练网络。如何在不丢失原始类的先前知识的情况下向该模型添加更多类?
我已经阅读了另一篇文章,该文章建议创建一个新模型,除了最后两层外,其他层均具有“基本模型”层。 (为什么要最后2个?)。并将新模型的最后2层权重设置为基本模型的最后2层权重。链接到帖子:如何为深度学习模型添加新类别
weights_training = base_model.layers[-2].get_weights()
new_model.layers[-2].set_weights(weights_training)
请注意,上面的代码在tensorflow中而不是tfjs中。
我能在tensorflow.js中做同样的事情吗?
解决方法
暂无找到可以解决该程序问题的有效方法,小编努力寻找整理中!
如果你已经找到好的解决方法,欢迎将解决方案带上本链接一起发送给小编。
小编邮箱:dio#foxmail.com (将#修改为@)