tensorflow保存 和 加载模型
2016-11-22 22:12
381 查看
1、
import tensorflow as tf
import numpy as np
# save to file
W = tf.Variable([[1,2,3],[4,5,6]],dtype = tf.float32,name='weight')
b = tf.Variable([[1,2,3]],dtype = tf.float32,name='biases')
init = tf.initialize_all_variables()
saver = tf.train.Saver()
with tf.Session() as sess:
sess.run(init)
save_path = saver.save(sess,"my_net/save_net.ckpt")
print ("save to path:",save_path)
说明:保存模型
2、
import tensorflow as tf
import numpy as np
W = tf.Variable(np.arange(6).reshape((2,3)),dtype = tf.float32,name='weight')
b = tf.Variable(np.arange(3).reshape((1,3)),dtype = tf.float32,name='biases')
saver = tf.train.Saver()
with tf.Session() as sess:
saver.restore(sess,"my_net/save_net.ckpt")
print ("weights:",sess.run(W))
print ("biases:",sess.run(b))
说明:加载模型
import tensorflow as tf
import numpy as np
# save to file
W = tf.Variable([[1,2,3],[4,5,6]],dtype = tf.float32,name='weight')
b = tf.Variable([[1,2,3]],dtype = tf.float32,name='biases')
init = tf.initialize_all_variables()
saver = tf.train.Saver()
with tf.Session() as sess:
sess.run(init)
save_path = saver.save(sess,"my_net/save_net.ckpt")
print ("save to path:",save_path)
说明:保存模型
2、
import tensorflow as tf
import numpy as np
W = tf.Variable(np.arange(6).reshape((2,3)),dtype = tf.float32,name='weight')
b = tf.Variable(np.arange(3).reshape((1,3)),dtype = tf.float32,name='biases')
saver = tf.train.Saver()
with tf.Session() as sess:
saver.restore(sess,"my_net/save_net.ckpt")
print ("weights:",sess.run(W))
print ("biases:",sess.run(b))
说明:加载模型
相关文章推荐
- TensorFlow保存和加载训练模型
- TensorFlow 深度学习框架(7)-- 变量管理及训练模型的保存与加载
- tensorflow模型保存与加载
- Tensorflow模型保存和加载
- TensorFlow 深度学习框架(7)-- 变量管理及训练模型的保存与加载
- tensorflow 保存和加载模型Saver的使用
- [Tensorflow之九]Tensorflow模型的保存与加载
- tensorflow中保存模型、加载模型做预测(不需要再定义网络结构)
- TensorFlow 深度学习框架(7)-- 变量管理及训练模型的保存与加载
- TensorFlow模型op的保存和加载(含演示代码)
- TensorFlow 深度学习框架(7)-- 变量管理及训练模型的保存与加载
- TensorFlow 深度学习框架(7)-- 变量管理及训练模型的保存与加载
- tensorflow初探---模型文件保存和加载
- 【tensorflow】保存模型、再次加载模型等操作
- tensorflow 保存和加载模型 -2
- 小白学Tensorflow之模型保存和加载
- TensorFlow 深度学习框架(7)-- 变量管理及训练模型的保存与加载
- Tensorflow加载预训练模型和保存模型的实例
- TensorFlow 深度学习框架(7)-- 变量管理及训练模型的保存与加载
- tensorflow学习之Faster R-CNN模型的保存与加载