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
目录
相关文章
|
24天前
|
机器学习/深度学习 PyTorch 算法框架/工具
通过实例学习Pytorch加载权重.load_state_dict()与保存权重.save()
通过实例学习Pytorch加载权重.load_state_dict()与保存权重.save()
13 0
base64转图片的时候Can't read input file!
如果不去掉编码中的图片头信息,就会报Can't read input file!。 一开始还以为是文件路径格式有问题,然后看了看我之前用的,就有一个区别就是没有去头信息。
|
机器学习/深度学习 搜索推荐 TensorFlow
tf_record_writer.py代码解释
这段代码是用来将电影评分数据集转换为 TensorFlow 训练所需的二进制 TFRecord 格式的。这里采用的是 MovieLens 数据集,其中包含了 1 百万个电影评分记录,用于推荐系统任务的训练和测试。 该代码主要分为几个部分:
Jmeter组件-Random CSV Data Set Config参数化CSV随机读取文件
Jmeter组件-Random CSV Data Set Config参数化CSV随机读取文件
Jmeter组件-Random CSV Data Set Config参数化CSV随机读取文件
|
移动开发
同样是保存模型,model.save()和model. save_weights ()有何区别
同样是保存模型,model.save()和model. save_weights ()有何区别
547 0
|
TensorFlow 算法框架/工具
TF学习——TF数据读取:TensorFlow中数据读这三张图片的5个epoch +把读取的结果重新存到read 文件夹中
TF学习——TF数据读取:TensorFlow中数据读这三张图片的5个epoch +把读取的结果重新存到read 文件夹中
TF学习——TF数据读取:TensorFlow中数据读这三张图片的5个epoch +把读取的结果重新存到read 文件夹中
|
TensorFlow API 算法框架/工具
|
Oracle 关系型数据库 Linux
[20180224]理解exp direct导出操作.txt
[20180224]理解exp direct导出操作.txt 1.环境: SCOTT@book> @ &r/ver1 PORT_STRING                    VERSION        BANNER -------------...
1053 0