老师,迁移后的模型如何保存?如何在其他的地方重复使用?
来源:12-5 迁移学习下的模型预测

zozo_zuo
2020-01-26
如题,课程里将模型分成了两部分(截断的mobilenet和自己编写的双层dense网络),如果想要将模型保存复用的话,是不是要将我们自己写的双层神经网络再单独保存成一个模型,然后在使用的时候,加载两次模型,先加载mobilenet模型,再加载我们自己写的双层神经网络模型,然后按照训练时的结构重新构建我们的网络?或者说有没有什么方法能将预训练模型的一部分与我们自己搭建的模型结合到一个模型中,然后将两部分一起保存,保存成一个完整的模型,这样在部署使用的时候只需要加载一个模型就可以了?
写回答
2回答
-
st2020
2020-02-02
预训练截断模型和我们新的模型结合在一起保存成一个模型 这是可以的吧 肯定是可以的 python 里就经常这样做
032020-02-02 -
lewis
2020-01-26
只能保存我们那个双层神经网络,引用的时候只能引用两个模型,这个没办法。
00
相似问题