小编Tha*_*Guy的帖子

无法在张量流中还原模型

我正在初始化保护程序

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)

python python-3.x tensorflow

5
推荐指数
0
解决办法
388
查看次数

标签 统计

python ×1

python-3.x ×1

tensorflow ×1