TensorlFlow中的一些坑
来源:互联网 发布:外贸进出口软件 编辑:程序博客网 时间:2024/06/05 21:13
model saver
- issue link:Saving and Restoring a trained LSTM in Tensor Flow
问题描述:在保存TensorFlow中的RNN/LSTM模型的时候,需要在LSTM模型建立之后再定义saver如:
### Model Training and Saving code### define the LSTM model code heresaver = tf.train.Saver()### train process heresaver.saver(sess, saver_path)
### Model Predict and Restore code### define the LSTM model code heresaver = tf.train.Saver()saver.restore(sess. saver_path)
事实上,如果是保存一般的variables也需要在定义了variable,如tf.get_variable()或者tf.Variable()中定义了之后,才能create Saver不然的话也会保存失败。下面就是一个例子:
saver = tf.train.Saver() tf.Variable([1], tf.float32) with tf.Session() as sess: sess.run(tf.global_variables_initializer()) saver.save(sess, path)
这样运行将会报错:
ValueError: No variables to save
而将Saver的定义位置换一下,就可以解决问题了。
tf.Variable([1], tf.float32) saver = tf.train.Saver() with tf.Session() as sess: sess.run(tf.global_variables_initializer()) saver.save(sess, path)
阅读全文
0 0
- TensorlFlow中的一些坑
- IOS7中的一些坑
- javascript中的一些坑
- UITableView中的一些坑
- ios9中的一些坑
- JavaScript 中的一些坑
- javascript中的一些坑
- Python 中的一些坑
- Android开发中的一些坑
- iOS开发中的一些坑
- Maven编译中的一些坑
- recyclerView使用中的一些坑
- Android开发中的一些坑
- shell编程中的一些坑
- JAVASCRIPT 使用中的一些坑
- 公众号建立过程中的一些坑
- 最近开发中的一些坑#1
- gitolite服务器部署中的一些坑
- css之dispaly:flex;多行多列
- visual studio 15.3 preview开始支持 MinGW and Cygwin
- hdu 1024 dp
- 安装坑爹依赖 之 node-sass
- Java——Arrays工具类介绍
- TensorlFlow中的一些坑
- python判断语句
- Difference (hdu5936——2016ccpc杭州站D题)
- Qt 学习之路 2(63):使用 QJson 处理 JSON
- leetcode刷题笔记
- poj 3225 Help with Intervals -线段树-延迟标记-区间交并补
- 674. Longest Continuous Increasing Subsequence
- mac-Linux安装MySQL
- crm开篇