tensorflow(三) 模型保存
来源:互联网 发布:js单选框选中触发事件 编辑:程序博客网 时间:2024/05/16 01:41
tensorflow最简单的保存与加载模型的方法是Saver对象(存放在tensorflow.train)。构造器给graph所有的变量,或者定义在列表中的变量,添加save和restore的操作,分别为保存和加载。变量保存在二进制的文件中,主要包含的是从变量名到tensor值的映射关系。
保存变量
通过下面的一段代码穿件Saver对象来管理模型中的变量(默认情况下是所有的变量,也可以自行选择)。
import tensorflow as tfv1 = tf.Variable(tf.random_normal([1,2]), name="v1")v2 = tf.Variable(tf.random_normal([2,3]), name="v2")init_op = tf.initialize_all_variables()saver = tf.train.Saver()with tf.Session() as sess: sess.run(init_op) saver_path = saver.save(sess, "/home/yang/data/model.ckpt") print "Model saved in file: ", saver_path
恢复变量
用同一个Saver对象来恢复变量,注意,当你从文件恢复变量是,不需要对它进行初始化,否则会报错。
import tensorflow as tfv1 = tf.Variable(tf.random_normal([1,2]), name="v1")v2 = tf.Variable(tf.random_normal([2,3]), name="v2")saver = tf.train.Saver()with tf.Session() as sess: saver.restore(sess, "/home/yang/data/model.ckpt") print "Model restored"
2 0
- tensorflow(三) 模型保存
- tensorflow(三) 模型保存
- TensorFlow学习系列(三):保存/恢复和混合多个模型
- TensorFlow学习系列(三):保存/恢复和混合多个模型
- tensorflow-模型保存和加载(二)
- Tensorflow进行保存模型
- tensorflow 模型保存问
- TensorFlow保存读取模型
- Tensorflow:模型保存/模型恢复?
- tensorflow保存 和 加载模型
- Tensorflow保存和读取模型
- Tensorflow模型保存与读取
- tensorflow-模型保存与读取
- tensorflow 模型保存与加载
- Tensorflow 保存和加载模型
- 保存和读取 TensorFlow 模型
- tensorflow模型保存文件分析
- TensorFlow之保存/恢复模型
- UVA11136 Hoax or what STL 或者 HEAP
- Leetcode 64 Minimum Path Sum
- ionic中自定义插件的问题
- 2016 ACM/ICPC Reginal Shengyang hdu 5893 List wants to travel(树链剖分 线段树区间更新真蛋疼)★
- 关于接口和多态
- tensorflow(三) 模型保存
- Unity平台的预处理及依赖于平台的编译
- 简单五子棋
- Java实现栈数据结构
- 组合数取模
- [JSPatch使用]OC转JS问题列表
- 使用JS实现选项卡和滑动门的效果
- poj3126 Prime Path BFS深度优先搜索
- Python 快速入门教程