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

Question

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

Answer 1

st2020 2020-02-02 20:52:04

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

• lewis #1

  或许可以，但是我没有这么做过，也没有见过类似的API和文档。

• st2020 回复 lewis #2

  官方例子就是这样的 https://www.tensorflow.org/tutorials/images/transfer_learning_with_hub

• st2020 #3

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

Answer 2

lewis 2020-01-26 21:18:31

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