82
edits
Changes
Jump to navigation
Jump to search
no edit summary
'''
if tf.test.is_gpu_available():
lstm = tf.contrib.cudnn_rnn.CudnnCompatibleLSTMCell(num_units)#num_units is the number of hidden units in the LSTM cell.
else:
lstm = tf.nn.rnn_cell.LSTMCell(num_units,forget_bias=1.0)
