从另一个python文件导入tf.train.Saver

Question

我使用以下代码在one.py文件中使用tf.train.Saver（） 。

saver = tf.train.Saver(tf.all_variables())
saver.save(sess,"checkpoint.data")

如何在另一个python文件中恢复checkpoint.data ？

我使用了以下代码，但是没有用。

from one import saver
import tensorflow as tf

with tf.Session() as sess:
    saver.restore(sess, "checkpoint.data")

Answer 1

检查点文件（即'checkpoint.data' ）没有为TensorFlow提供足够的信息来重构模型结构。 在第二个程序中，您需要重建在第一个程序中使用的相同的TensorFlow图。 有几种方法可以执行此操作：

• 将模型构建代码提取到Python函数中，并在每个程序中创建tf.train.Saver之前调用它。
• 使用saver.export_meta_graph()在第一个程序中写出图形结构以及检查点，并在第二个程序中使用tf.train.import_meta_graph()导入图形结构（并创建适当配置的tf.train.Saver实例）。程序。