从先前的检查点还原Tensorflow模型

时间:2018-10-22 13:39:40

标签: python tensorflow neural-network

我正在通过对模型进行两次训练(每次100次)来调整我的模型,使其达到200次。 100 eps的第一次运行完成,现在第二次运行正在进行。我的问题是我要在完成第100集的第一次运行后恢复模型。现在我已经开始第二次运行,它已经覆盖了检查点文件。

我碰到this post,它说tensorflow保留最后5个模型。如何从第一次运行后创建的上一个检查点检索模型参数?

1 个答案:

答案 0 :(得分:0)

Tensorflow可以在一次运行中保存最好的5个模型,您已经开始了另一个新的运行,恐怕您的模型会丢失,您将不得不对其进行重新训练,凭经验法则始终保持模型的动态命名,因此它们不会不能替换,例如:

model_name = 'Model_{:d}'.format(episode_number) + '.ckpt'