我正在初始化保护程序
self.saver = tf.train.Saver()
Run Code Online (Sandbox Code Playgroud)
在图形构造的末尾。然后我用保存模型
self.saver.save(sess, self.data_batch.params.LOG_DIR + "model", global_step=epoch)
Run Code Online (Sandbox Code Playgroud)
最后,我尝试使用还原它:
with tf.Session(graph=self.graph, config=self.proto_config) as sess:
if restore:
self.saver.restore(sess, "./" + self.data_batch.params.LOG_DIR)
else:
self.init.run()
Run Code Online (Sandbox Code Playgroud)
我的文件夹结构如下所示:
-Main directory
--main
--Logs/
---checkpoint
---events.out.tfevents.1528047461.Ublion18
---model-0.data-00000-of-00001
---model-0.index
---model-0.meta
---params
Run Code Online (Sandbox Code Playgroud)
错误如下:
Caused by op 'save/RestoreV2', defined at:
File "/home/dizzi/Work/PycharmProjects/NLP/HomeWork2/main.py", line 22, in <module>
model.init_graph()
File "/home/dizzi/Work/PycharmProjects/NLP/HomeWork2/Classes/Model.py", line 293, in init_graph
self.saver = tf.train.Saver()
File "/home/dizzi/anaconda3/envs/NLP/lib/python3.6/site-packages/tensorflow/python/training/saver.py", line 1311, in __init__
self.build()
File "/home/dizzi/anaconda3/envs/NLP/lib/python3.6/site-packages/tensorflow/python/training/saver.py", line 1320, in build
self._build(self._filename, build_save=True, build_restore=True)
File "/home/dizzi/anaconda3/envs/NLP/lib/python3.6/site-packages/tensorflow/python/training/saver.py", line 1357, in …Run Code Online (Sandbox Code Playgroud)