在Tensorflow中加载已保存的RNN

时间:2017-04-17 16:02:48

标签: python tensorflow

我刚从TensorFlow那里学习本教程。

提供完整的分步指南here

源代码为here

现在我想做更多的事情。我想保存训练好的RNN并将其加载到另一个python脚本中,这样我就可以向RNN提供输入,它将生成下一个可能的单词。

要保存模型,我运行了命令:

cd models/tutorials/rnn/ptb
python ptb_word_lm.py --data_path=/tmp/simple-examples/data/ --model=small --save_path=~/Desktop/SavedRNN/

然后我得到了一个带有这些条目的检查点文件:

model_checkpoint_path: "/home/piyush/Documents/college courses/year 3/sem 6/Information Retrieval/project2/-30199"
all_model_checkpoint_paths: "/home/piyush/Documents/college courses/year 3/sem 6/Information Retrieval/project2/model.ckpt-24745"
all_model_checkpoint_paths: "/home/piyush/Documents/college courses/year 3/sem 6/Information Retrieval/project2/model.ckpt-26220"
all_model_checkpoint_paths: "/home/piyush/Documents/college courses/year 3/sem 6/Information Retrieval/project2/model.ckpt-27762"
all_model_checkpoint_paths: "/home/piyush/Documents/college courses/year 3/sem 6/Information Retrieval/project2/model.ckpt-29237"
all_model_checkpoint_paths: "/home/piyush/Documents/college courses/year 3/sem 6/Information Retrieval/project2/-30199" 

我有3个类型的9个文件

models.ckpt.<some-number>.data-00000-of-00001
models.ckpt.<some-number>.meta
models.ckpt.<some-number>.index

现在我无法在tensorflow中找到任何加载此模型的文档。我遇到的所有示例都只使用tf.saver只有一个检查点。

我猜,它与supervisor有关。但我不确定。

现在我想要执行的步骤是:

  1. 在TensorFlow(python)
  2. 中加载已保存的模型
  3. 使用我自己的输入运行模型。

1 个答案:

答案 0 :(得分:0)

要加载模型,请在加载模型时将检查点路径传递到<some_number>到保护程序(这应该是在构建图形之后,重用所提供示例中的代码)。