返回信息流求问论坛上的小伙伴们,
我之前训练了一个LSTM的tensorflow模型L1(参数: LSTM1,Wy1等),现在想训练一个新的模型L2,但只需要restoreL1的参数LSTM1,那我该怎么设置saver和restore呢?求一份相关code
[ema1]
这是一条镜像帖。来源:北邮人论坛 / ml-dm / #20692同步于 2016/7/24
该镜像源已超过 30 天没有更新,可能在源站已被删除。
ML_DM机器人发帖
tensorflow restore问题
sophieag
2016/7/24镜像同步3 回复
订阅后,新回复会通过你的通知中心匿名送达。
3 条回复
直觉告诉我应该是restore完进行assign
https://www.tensorflow.org/versions/r0.9/api_docs/python/state_ops.html#Saver
【 在 damcy 的大作中提到: 】
: 直觉告诉我应该是restore完进行assign
: https://www.tensorflow.org/versions/r0.9/api_docs/python/state_ops.html#Saver
额 我找到一种方法 我先定义saver([W1..])中间为模型L2会用到的参数,然后restore,能跑。
但是有个很重要的问题,我训练LSTM,而tensorflow中LSTM()算是什么呢?变量么?就LSTM里面的参数我提不出来
是这样的。LSTM里面的参数也是用get_variable定义的,不过不像自定义的知道名字,所以在模型搭建完之后用tf.trainable_variables可以看一下创建了哪些变量,然后取得时候你就知道名字啦。一般创建的第一个LSTM,参数名字是'RNN/LSTMCell/W_0' & 'RNN/LSTMCell/B',用tf.trainable_variables可以得到graph的所有variables
【 在 sophieag 的大作中提到: 】
:
: 额 我找到一种方法 我先定义saver([W1..])中间为模型L2会用到的参数,然后restore,能跑。
: 但是有个很重要的问题,我训练LSTM,而tensorflow中LSTM()算是什么呢?变量么?就LSTM里面的参数我提不出来