從另一個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實例）。程序。