tf:save-restore保存与读取

简介:

saver实例代码:

[python]  view plain  copy
  1. ## Save to file  
  2. # remember to define the same dtype and shape when restore  
  3. W = tf.Variable([[1,2,3],[3,4,5]], dtype=tf.float32, name='weights')  
  4. b = tf.Variable([[1,2,3]], dtype=tf.float32, name='biases')  
  5.   
  6. init= tf.global_variables_initialize()  
  7.   
  8. saver = tf.train.Saver()  
  9.   
  10. with tf.Session() as sess:  
  11.    sess.run(init)  
  12.    save_path = saver.save(sess, "my_net/save_net.ckpt")  
  13.    print("Save to path: ", save_path)  

restore实例代码:

[html]  view plain  copy
  1. # restore variables  
  2. # redefine the same shape and same type for your variables  
  3. W = tf.Variable(np.arange(6).reshape((2, 3)), dtype=tf.float32, name="weights")  
  4. b = tf.Variable(np.arange(3).reshape((1, 3)), dtype=tf.float32, name="biases")  
  5.   
  6. # not need init step  
  7.   
  8. saver = tf.train.Saver()  
  9. with tf.Session() as sess:  
  10.     saver.restore(sess, "my_net/save_net.ckpt")  
  11.     print("weights:", sess.run(W))  
  12.     print("biases:", sess.run(b))  

转自:http://blog.csdn.net/smf0504/article/details/56666305
目录
相关文章
|
5月前
磁盘文件"test"中保存
【7月更文挑战第9天】磁盘文件"test"中保存。
47 13
|
7月前
|
机器学习/深度学习 PyTorch 算法框架/工具
通过实例学习Pytorch加载权重.load_state_dict()与保存权重.save()
通过实例学习Pytorch加载权重.load_state_dict()与保存权重.save()
88 0
Jmeter组件-Random CSV Data Set Config参数化CSV随机读取文件
Jmeter组件-Random CSV Data Set Config参数化CSV随机读取文件
Jmeter组件-Random CSV Data Set Config参数化CSV随机读取文件
|
TensorFlow 算法框架/工具
TF学习——TF数据读取:TensorFlow中数据读这三张图片的5个epoch +把读取的结果重新存到read 文件夹中
TF学习——TF数据读取:TensorFlow中数据读这三张图片的5个epoch +把读取的结果重新存到read 文件夹中
TF学习——TF数据读取:TensorFlow中数据读这三张图片的5个epoch +把读取的结果重新存到read 文件夹中
|
存储 C语言 C++
MAT文件打开方法汇总及其他操作
MAT文件打开方法汇总及其他操作
2689 0
|
移动开发
同样是保存模型,model.save()和model. save_weights ()有何区别
同样是保存模型,model.save()和model. save_weights ()有何区别
645 0
|
Oracle 关系型数据库 Linux
[20180224]理解exp direct导出操作.txt
[20180224]理解exp direct导出操作.txt 1.环境: SCOTT@book> @ &r/ver1 PORT_STRING                    VERSION        BANNER -------------...
1087 0