我正在关注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.