导出和导入tensorflow网络在状态评估中的应用

2024-04-24 07:08:26 发布

您现在位置:Python中文网/ 问答频道 /正文

我正在用tensorflow编写一个神经网络,我希望能够导出我最终训练的网络,并将其导入另一个程序中进行游戏。我发现了多个论坛帖子,比如: Tensorflow: How to use a trained model in a application?Tensorflow: how to save/restore a model?

我在tf的文档中也看到了他们使用估计器来保存模型,但我不确定这是否是我想要的,以及如何应用它。你知道吗

但是他们谈论的是导出整个会话并将其导入到应用程序中,然后使用会话.run,但据我所知,这需要输入预测输出,并将在我的网络上运行另一个训练步骤。我不想继续训练我的网络-它结束了-我现在只想评估游戏给我的特定状态。你知道吗

提前感谢您的帮助。你知道吗


Tags: toin程序网络游戏modelapplicationuse
1条回答
网友
1楼 · 发布于 2024-04-24 07:08:26

据我所知,有两种方法。你知道吗

  1. 检查点文件(元图)
  2. 保存的模型

savedmodel非常方便,但是学习曲线比检查点文件要高。你可以检查这个tutorial

而导入模型并不是继续运行的训练,它基本上是还原你所学的所有变量。你知道吗

相关问题 更多 >