Tensorflow保护程序还原。变量值似乎已经改变。

时间:2017-06-01 15:14:10

标签: python tensorflow

我正在使用tensorflow进行训练,以创建一个训练有素的模型,然后用

保存到检查点
saver = tf.train.Saver(max_to_keep=1)
saver.save(sess, checkpointPath, global_step=global_step+1000)

稍后我构建相同的模型并使用

初始化它
ckpt = tf.train.get_checkpoint_state(checkpointDir)    
saver = tf.train.Saver(max_to_keep=1)
saver.restore(sess, ckpt.model_checkpoint_path)

当我使用这个恢复的网络运行几个步骤时,如果我继续使用原始网络,我会得到完全不同的结果。我希望它们是一样的。我想弄清楚原因。

有人有什么建议吗?我相信我正在保存所有变量(尽管可能不是)。

我想设置一个测试,声明所有变量在恢复的模型中与保存的模型中的相同,但无法解决如何执行此操作。有没有人有代码片段来显示这个?

我想使用我保存的模型来加载经过训练的预测器。

0 个答案:

没有答案