TensorLayer 当 LSTM 设为 state_is_tuple 时该如何迭代。

2016年8月10日作。

State 不是 Tuple 时,代码如下。一个epoch的最开始,我们要把一个 全零array,输入到 initial_state 中,然后输出 finial_state 覆盖到 state 中,然后执行下一个迭代。

但是当LSTM设置为 state is tuple, TensorFlow是返回一个LSTMStateTuple的,是不可以通过eval() 来得到参数的,而且也不能被feed!!

下面的代码中,tl.layers.initialize_rnn_state 使用eval 得到initial state 的全零array。

来源:https://github.com/zsdonghao/tensorlayer/blob/master/tutorial_ptb_lstm.py


 
  state1 = tl.layers.initialize_rnn_state(lstm1.initial_state)
  state2 = tl.layers.initialize_rnn_state(lstm2.initial_state)
  for step, (x, y) in enumerate(tl.iterate.ptb_iterator(train_data,
  batch_size, num_steps)):
  feed_dict = {input_data: x, targets: y,
  lstm1.initial_state: state1,
  lstm2.initial_state: state2,
  }
  # For training, enable dropout
  feed_dict.update( network.all_drop )
  _cost, state1, state2, _ = sess.run([cost,
  lstm1.final_state,
  lstm2.final_state,
  train_op],
  feed_dict=feed_dict
  )
  costs += _cost; iters += num_steps
 



为了解决这个问题,当State 是Tuple 时 ,tl.layers.initialize_rnn_state 会返回一个 tuple 里面有对应 cell 和 hidden state 的全零 array。

然后,当feed in 时,还要分别对 cell 和 hidden state feed in。

来源:https://github.com/zsdonghao/tensorlayer/blob/master/tutorial_ptb_lstm_state_is_tuple.py


 
  state1 = tl.layers.initialize_rnn_state(lstm1.initial_state)
  state2 = tl.layers.initialize_rnn_state(lstm2.initial_state)
  for step, (x, y) in enumerate(tl.iterate.ptb_iterator(train_data,
  batch_size, num_steps)):
  feed_dict = {input_data: x, targets: y,
  lstm1.initial_state.c: state1[0],
  lstm1.initial_state.h: state1[1],
  lstm2.initial_state.c: state2[0],
  lstm2.initial_state.h: state2[1],
  }
  # For training, enable dropout
  feed_dict.update( network.all_drop )
  _cost, state1_c, state1_h, state2_c, state2_h, _ = \
  sess.run([cost,
  lstm1.final_state.c,
  lstm1.final_state.h,
  lstm2.final_state.c,
  lstm2.final_state.h,
  train_op],
  feed_dict=feed_dict
  )
  state1 = (state1_c, state1_h)
  state2 = (state2_c, state2_h)

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值