From 78bb26286e2a8486bf8b1e4e68c4ce0de9ad002a Mon Sep 17 00:00:00 2001 From: Federico Fontana Date: Thu, 2 May 2019 21:19:27 +0100 Subject: [PATCH] Replaced discontinued rnn_cell.BasicLSTMCell with rnn_cell.LSTMCell (#4703) * Fixed bug in Dirichlet (#4440) * Replaced deprecated rnn_cell.BasicLSTMCell with rnn_cell.LSTMCell --- python/ray/rllib/models/lstm.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/python/ray/rllib/models/lstm.py b/python/ray/rllib/models/lstm.py index b3fb557b8..18f141d09 100644 --- a/python/ray/rllib/models/lstm.py +++ b/python/ray/rllib/models/lstm.py @@ -56,7 +56,7 @@ class LSTM(Model): last_layer = add_time_dimension(features, self.seq_lens) # Setup the LSTM cell - lstm = rnn.BasicLSTMCell(cell_size, state_is_tuple=True) + lstm = tf.nn.rnn_cell.LSTMCell(cell_size, state_is_tuple=True) self.state_init = [ np.zeros(lstm.state_size.c, np.float32), np.zeros(lstm.state_size.h, np.float32)