experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

deprecated keep_prob removed from dropout in grappler/hierarchical_controller.py

Browse Source
This commit is contained in:
Siju Samuel 2019-07-05 09:57:37 +05:30
parent 0fe684b5eb
commit e6a655127a
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
tensorflow/python/grappler/hierarchical_controller.py
View File

@ -883,7 +883,7 @@ class HierarchicalController(Controller):
actions.read(i - 1)) actions.read(i - 1))
) )
if self.hparams.keep_prob is not None: if self.hparams.keep_prob is not None:
signal = nn_ops.dropout(signal, self.hparams.keep_prob) signal = nn_ops.dropout(signal, rate=(1 - self.hparams.keep_prob))
next_c, next_h = lstm(signal, prev_c, prev_h, w_lstm, forget_bias) next_c, next_h = lstm(signal, prev_c, prev_h, w_lstm, forget_bias)
query = math_ops.matmul(next_h, attn_w_2) query = math_ops.matmul(next_h, attn_w_2)
query = array_ops.reshape( query = array_ops.reshape(