From 370ae0b1e4e37ff3871e4888e2a7395adc2a1195 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?=E5=AE=89=E5=85=B4=E4=B9=90-siler?= Date: Wed, 22 Feb 2017 12:03:36 +0800 Subject: [PATCH] fix BasicLSTMCell for TF1.0 BasicLSTMCell should be tf.contrib.rnn.BasicLSTMCell in TF1.0 --- docs/user/tutorial.rst | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/docs/user/tutorial.rst b/docs/user/tutorial.rst index a4ecc9838..4e6bf25bf 100755 --- a/docs/user/tutorial.rst +++ b/docs/user/tutorial.rst @@ -1086,7 +1086,7 @@ To understand the PTB tutorial, you can also read `TensorFlow PTB tutorial if is_training: network = tl.layers.DropoutLayer(network, keep=keep_prob, name='drop1') network = tl.layers.RNNLayer(network, - cell_fn=tf.nn.rnn_cell.BasicLSTMCell, + cell_fn=tf.contrib.rnn.BasicLSTMCell, cell_init_args={'forget_bias': 0.0}, n_hidden=hidden_size, initializer=tf.random_uniform_initializer(-init_scale, init_scale), @@ -1097,7 +1097,7 @@ To understand the PTB tutorial, you can also read `TensorFlow PTB tutorial if is_training: network = tl.layers.DropoutLayer(network, keep=keep_prob, name='drop2') network = tl.layers.RNNLayer(network, - cell_fn=tf.nn.rnn_cell.BasicLSTMCell, + cell_fn=tf.contrib.rnn.BasicLSTMCell, cell_init_args={'forget_bias': 0.0}, n_hidden=hidden_size, initializer=tf.random_uniform_initializer(-init_scale, init_scale),