【问题】Tensorflow怎么保存RNN模型。。

2017/3/17镜像同步11 回复

最近用tf写LSTM，但不知道训练完不知道怎么保存模型。。 Tensorflow写cnn的时候可以用save，restore来保存，读取session中的W,b参数。但是写LSTM的时候用到了tensorflow库中的lstm_cell和tf.nn.dynamic_rnn，里面包括了forget gate/input gate相关的参数，那么这些参数是怎么保存呢？。。试过直接用restore来保存貌似不对。。

订阅后，新回复会通过你的通知中心匿名送达。

9 条回复

LiuAiting机器人#1 · 2017/3/17

也是那样保存

moses机器人#2 · 2017/3/18

楼主用 LSTM 做什么？我也在用，希望可以交流一下

t2396156机器人#3 · 2017/3/18

保存方法一样

stamdlee机器人#4 · 2017/3/20

棒棒！【在 LiuAiting 的大作中提到: 】 : 也是那样保存

liuyou6046机器人#5 · 2017/3/20

【在 WindSword 的大作中提到: 】 : 最近用tf写LSTM，但不知道训练完不知道怎么保存模型。。 : Tensorflow写cnn的时候可以用save，restore来保存，读取session中的W,b参数。 : 但是写LSTM的时候用到了tensorflow库中的lstm_cell和tf.nn.dynamic_rnn，里面包括了forget gate/input gate相关的参数，那么这些参数是怎么保存呢？。。 : ................... 你在windows用的tensorflow吗

WindSword机器人#6 · 2017/3/21

目前只是写写基础模型跑跑kaggle练习一下~ 打算过段时间有空再看论文写点程序。【在 moses 的大作中提到: 】 : 楼主用 LSTM 做什么？我也在用，希望可以交流一下

WindSword机器人#7 · 2017/3/21

我是在mac上用的。【在 liuyou6046 的大作中提到: 】 : 你在windows用的tensorflow吗

liuyou6046机器人#8 · 2017/3/21

【在 WindSword 的大作中提到: 】 : 我是在mac上用的。 : 在windows上用的已经哭了

WindSword机器人#9 · 2017/3/21

是直接save(session),restore(session)就行么.. 我看文档说，restore是按变量名和tensor值的映射关系保存的。层与层之间的w，b可以自己定义变量名。但是在LSTMcell中的变量貌似都是在调用的函数里面，没有自己定义变量名。这样restore的时候也可以对应上么？【在 LiuAiting 的大作中提到: 】 : 也是那样保存