我刚从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
有关。但我不确定。
现在我想要执行的步骤是:
答案 0 :(得分:0)
要加载模型,请在加载模型时将检查点路径传递到<some_number>
到保护程序(这应该是在构建图形之后,重用所提供示例中的代码)。