老师,迁移后的模型如何保存?如何在其他的地方重复使用?

来源:12-5 迁移学习下的模型预测

zozo_zuo

2020-01-26

如题,课程里将模型分成了两部分(截断的mobilenet和自己编写的双层dense网络),如果想要将模型保存复用的话,是不是要将我们自己写的双层神经网络再单独保存成一个模型,然后在使用的时候,加载两次模型,先加载mobilenet模型,再加载我们自己写的双层神经网络模型,然后按照训练时的结构重新构建我们的网络?或者说有没有什么方法能将预训练模型的一部分与我们自己搭建的模型结合到一个模型中,然后将两部分一起保存,保存成一个完整的模型,这样在部署使用的时候只需要加载一个模型就可以了?

写回答

2回答

st2020

2020-02-02

预训练截断模型和我们新的模型结合在一起保存成一个模型 这是可以的吧 肯定是可以的 python 里就经常这样做

0
3
st2020
tfjs 不太清楚 估计也是可以的 做一个 image classification 要用两个模型有点 confusing, 一般都是直接一个就行了 特殊情况出于性能(比如base model 的结果缓存给多个 work model 用)或者其他因素考虑才会考虑用一个 base model(截断模型)加多个 work model
2020-02-02
共3条回复

lewis

2020-01-26

只能保存我们那个双层神经网络,引用的时候只能引用两个模型,这个没办法。

0
0

JavaScript玩转机器学习-Tensorflow.js项目实战

机器学习理论知识+Tensorflow.js实战开发

644 学习 · 189 问题

查看课程