导出和导入tensorflow网络以评估应用程序中的状态

时间:2017-09-21 06:24:10

标签: python tensorflow neural-network

我正在写一个tensorflow中的神经网络,我希望能够导出我最后训练过的网络并将其导入另一个程序来玩游戏。我找到了多个论坛帖子,如: Tensorflow: How to use a trained model in a application? Tensorflow: how to save/restore a model?

我还在tf文档中看到他们使用估算器来保存模型,但我不确定这是否是我正在寻找的以及如何应用它。

但那些谈论导出整个会话并将其导入应用程序并使用Session.run,但据我所知,它需要输入预测输出,并将在我的网络上运行另一个训练步骤。我不想继续训练我的网络 - 它已经完成了 - 我现在想评估一下游戏给我的特定状态。

提前感谢您提供的任何帮助。

1 个答案:

答案 0 :(得分:0)

据我所知,有两种方法可以做到。

  1. 检查点文件(元图)
  2. savedmodel
  3. savedmodel非常方便,但学习曲线高于检查点文件。你可以查看tutorial

    并且导入模型不是继续运行训练,它基本上是恢复你学到的所有变量。