模型转换的问题

来源：9-3 Keras模型转化为SavedModel

慕移动4519168

2020-02-12

请问老师，如何将Keras（非tf.keras）训练的模型用tf. saved_model. save保存为pb类型的模型呢？

写回答

1回答

正十七

2020-02-26

已采纳

同学你好，这是个好问题，之前我没有考虑到过。

我觉得可以先试试，keras保存成 .h5文件，然后tf.keras载入后再保存成saved_model。

如果不可以，一个笨办法在一个程序文件中把keras的每层的参数都取出来作为tf.keras模型的初始化参数，然后再保存。

Google老师亲授 TensorFlow2.0 入门到进阶

Tensorflow2.0实战—以实战促理论的方式学习深度学习

1849 学习 · 896 问题

相似问题

回答 1

回答 2

回答 1

回答 1

回答 1