无法保存/导出并加载使用急切执行的keras模型

时间:2019-06-17 18:19:50

标签: tensorflow keras tf.keras

我正在关注RNN文本生成教程,并渴望逐行执行。我已经使用自己的数据集训练了模型,并保存了一个低损耗检查点。我能够加载权重并生成文本,但是我想导出/保存模型,以便我可以学习如何使用flask进行部署。但是我不知道怎么做。我使用的版本是“ 1.14.0-rc1”。

教程:https://www.tensorflow.org/tutorials/sequences/text_generation

我已经能够将模型另存为HDF5文件,但无法加载。我还禁用了急切执行的功能,但这会在以后运行代码时引起问题。我尝试了以下内容以及其他一些片段,但这些片段也没有带来任何好处:

new_model = keras.models.load_model("/content/gdrive/My Drive/ColabNotebooks/ckpt4/my_model.h5")

我怎么得到

RuntimeError: tf.placeholder() is not compatible with eager execution.

最后我在另一篇文章中找到了这个,并尝试了一下,但是遇到另一个错误:

tf.saved_model.save(model, "/content/gdrive/My Drive/Colab Notebooks/ckpt4/my_model.h5")

错误:

AssertionError: Tried to export a function which references untracked object Tensor("StatefulPartitionedCall/args_2:0", shape=(), dtype=resource).TensorFlow objects (e.g. tf.Variable) captured by functions must be tracked by assigning them to an attribute of a tracked object or assigned to an attribute of the main object directly.

0 个答案:

没有答案