Tensorflow中变量保存与恢复
来源:互联网 发布:新概念英语软件知乎 编辑:程序博客网 时间:2024/05/16 08:28
Saver
Tensorflow中,用 tensorflow.train.Saver来保存、恢复变量。
保存变量
tf.train.Saver.save(sess, save_path, global_step=None, latest_filename=None, meta_graph_suffix='meta', write_meta_graph=True, write_state=True)
#————————————————————例子————————————————————————import tensorflow as tf# 创建两个变量v1= tf.Variable(tf.random_normal([784, 200], stddev=0.35), name="v1")v2= tf.Variable(tf.zeros([200]), name="v2")# 添加用于初始化变量的节点init_op = tf.global_variables_initializer()# Create a saver.saver = tf.train.Saver(tf.global_variables())# 运行,保存变量sess = tf.Session()saver.save(sess,'my-model')
Saver可以使用提供的计数器自动为checkpoint文件编号。这使得在训练模型时在不同的步骤保留多个检查点。在save()方法中传递可选的global_step参数,可以对checkpoint文件进行编号
saver.save(sess, 'my-model', global_step=0) ==> filename: 'my-model-0'...saver.save(sess, 'my-model', global_step=1000) ==> filename: 'my-model-1000'
#——————————————————例子————————————————————————import tensorflow as tf# 创建两个变量v1= tf.Variable(tf.random_normal([784, 200], stddev=0.35), name="v1")v2= tf.Variable(tf.zeros([200]), name="v2")# 添加用于初始化变量的节点init_op = tf.global_variables_initializer()# Create a saver.saver = tf.train.Saver(tf.global_variables())# 运行图,打开会话,每1000次保存一个模型sess = tf.Session()for step in range(10000): sess.run(init_op) if step % 1000 == 0: saver.save(sess, base_path+'my-model', global_step=step)
运行结果:
恢复变量
tf.train.Saver.restore(sess, save_path)
#————————————————例子—————————————————————— sess.run(tf.global_variables_initializer()) saver = tf.train.Saver(tf.global_variables()) #arg:获取最近一次保存的变量文件名称 module_file = tf.train.latest_checkpoint('my-model') print(module_file) saver.restore(sess, module_file)
欲了解saver更详细的内容,请戳tensorflow.saver
0 0
- Tensorflow中变量保存与恢复
- Tensorflow-变量保存与导入
- 【深度学习】Tensorflow模型保存与恢复
- tensorflow 模型的保存与恢复(Saver)
- tensorflow中模型的保存和恢复
- tensorflow 变量简单存储与恢复
- tensorflow中变量的保存和加载
- tensorflow保存部分变量
- 16、TensorFLow 模型参数的保存与恢复
- tensorflow 1.0 学习:模型的保存与恢复(Saver)
- TensorFlow之保存/恢复模型
- Tensorflow:如何保存/恢复模型?
- Tensorflow:模型保存/模型恢复?
- tensorflow学习笔记(五):变量保存与导入
- iOS中cookie的保存与恢复
- tensorflow saver 保存和恢复指定 tensor
- Tensorflow学习(6)模型的保存与恢复(saver)
- Tensorflow学习: 保存变量和网络
- Docker 使用教程
- 掌握Bitmaps
- python urllib2详解及实例
- MATLAB 7.0 学习入门 -- (练习三 多项式求根)
- 依赖Android AAR的Cordova插件制作
- Tensorflow中变量保存与恢复
- License Key Formatting
- 编写高质量OC代码52建议总结:13.用“方法调配技术”调试“黑盒方法”
- C#如何调用以管理员身份运行的cmd命令提示符
- robotframework封装关键字用于控制ats(apache traffic server)实现ats自动化测试
- Rectjs问题总结
- 了解生成对抗网络GAN
- JSP常用标签——JSTL标签和EL表达式
- Java DOM解析XML的几个例子