From e6a655127a54f0f4361c7b44fd2ac4f125e85f29 Mon Sep 17 00:00:00 2001 From: Siju Samuel Date: Fri, 5 Jul 2019 09:57:37 +0530 Subject: [PATCH] deprecated keep_prob removed from dropout in grappler/hierarchical_controller.py --- tensorflow/python/grappler/hierarchical_controller.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tensorflow/python/grappler/hierarchical_controller.py b/tensorflow/python/grappler/hierarchical_controller.py index c8f2f4245bb..e39988d96b5 100644 --- a/tensorflow/python/grappler/hierarchical_controller.py +++ b/tensorflow/python/grappler/hierarchical_controller.py @@ -883,7 +883,7 @@ class HierarchicalController(Controller): actions.read(i - 1)) ) 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) query = math_ops.matmul(next_h, attn_w_2) query = array_ops.reshape(