解决tensorflow中报错NotFoundError (see above for traceback): Key v1_1 not found in checkpoint问题
tensorflow
一个面向所有人的开源机器学习框架
项目地址:https://gitcode.com/gh_mirrors/te/tensorflow
免费下载资源
·
- 针对TensorFlow实战框架Chp5,P113面问题
在 saver.restore()加载的前面,需要添加 tf.reset_default_graph(),作用是 清除默认图的堆栈,并设置全局图为默认图
保存文件
# -*- coding: utf-8 -*-
"""
Created on Sat Jun 30 16:17:26 2018
@author: muli
"""
import tensorflow as tf
v1 = tf.Variable(tf.constant(1.0,shape=[1]),name="v1")
v2 = tf.Variable(tf.constant(2.0,shape=[1]),name="v2")
result = v1 + v2
# 保存训练好的数据
saver = tf.train.Saver()
with tf.Session() as sess:
sess.run(tf.global_variables_initializer())
saver.save(sess, "Saved_model/model.ckpt")
print("已保存完毕!")
- 加载文件
# -*- coding: utf-8 -*-
"""
Created on Sat Jun 30 16:23:46 2018
@author: muli
"""
import tensorflow as tf
# 清除默认图的堆栈,并设置全局图为默认图
tf.reset_default_graph()
v1 = tf.Variable(tf.constant(1.0,shape=[1]),name="v1")
v2 = tf.Variable(tf.constant(2.0,shape=[1]),name="v2")
result = v1 + v2
# 保存训练好的数据
saver = tf.train.Saver()
with tf.Session() as sess:
print("正在加载...")
saver.restore(sess, "./Saved_model/model.ckpt")
print(sess.run(result))
- 直接加载已经持久化的图
# -*- coding: utf-8 -*-
"""
Created on Sat Jun 30 16:29:50 2018
@author: muli
"""
import tensorflow as tf
# 清除默认图的堆栈,并设置全局图为默认图
tf.reset_default_graph()
saver = tf.train.import_meta_graph("./Saved_model/model.ckpt.meta")
with tf.Session() as sess:
saver.restore(sess, "./Saved_model/model.ckpt")
print(sess.run(tf.get_default_graph().get_tensor_by_name("add:0")))
GitHub 加速计划 / te / tensorflow
184.55 K
74.12 K
下载
一个面向所有人的开源机器学习框架
最近提交(Master分支:2 个月前 )
a49e66f2
PiperOrigin-RevId: 663726708
3 个月前
91dac11a
This test overrides disabled_backends, dropping the default
value in the process.
PiperOrigin-RevId: 663711155
3 个月前
更多推荐
已为社区贡献6条内容
所有评论(0)