使用tensorflow保存后文件不存在

时间:2017-11-18 15:53:36

标签: python tensorflow keras

我使用tf.train.Saver('./model.ckpt')保存了我的模型,当我转到本地目录时,我找到了名为model.ckpt.indexmodel.ckpt.metamodel.ckpt.data-00000-of-00001的文件,但没有{{1 }}。结果,我无法恢复模型。谁知道我做错了什么?这是我的代码

model.ckpt

1 个答案:

答案 0 :(得分:2)

* .meta文件包含MetaGraph,您可以使用以下命令导入:

saver = tf.train.import_meta_graph("model.ckpt.meta")

然后您可以恢复图表的变量。

saver.restore(sess, "model.ckpt")

您可以通过调用tf.train.export_meta_graph将其他模型数据保存到元图中。

或者,您可以使用SavedModel存储应用程序的模型,SavedModel可以包含多个MetaGraph。文档为here