wassname/ray
1
0
Fork 0
mirror of https://github.com/wassname/ray.git synced 2026-06-28 00:29:38 +08:00
Code Issues Packages Projects Releases Wiki Activity

Replaced discontinued rnn_cell.BasicLSTMCell with rnn_cell.LSTMCell (#4703)

Browse Source 
* Fixed bug in Dirichlet (#4440)

* Replaced deprecated rnn_cell.BasicLSTMCell with rnn_cell.LSTMCell
This commit is contained in:
Federico Fontana
2019-05-02 21:19:27 +01:00
committed by Eric Liang
parent f87235f232
commit 78bb26286e
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
python/ray/rllib/models/lstm.py
+1 -1
View File
@@ -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)