Deprecated tf.random_normal to tf.random.normal in lstm README

This commit is contained in:
Siju Samuel 2019-07-08 11:08:41 +05:30
parent 7a193942cd
commit c5f4fe9b95

View File

@ -71,7 +71,7 @@ tflite_model = converter.convert() # You got a tflite model!
+ tf.lite.experimental.nn.TFLiteLSTMCell(
self.num_lstm_units, forget_bias=0))
# Weights and biases for output softmax layer.
out_weights = tf.Variable(tf.random_normal([self.units, self.num_class]))
out_weights = tf.Variable(tf.random.normal([self.units, self.num_class]))
@@ -67,7 +67,7 @@ class MnistLstmModel(object):
lstm_cells = tf.nn.rnn_cell.MultiRNNCell(lstm_layers)
# Note here, we use `tf.lite.experimental.nn.dynamic_rnn` and `time_major`
@ -170,7 +170,7 @@ class MnistLstmModel(object):
tf.lite.experimental.nn.TFLiteLSTMCell(
self.num_lstm_units, forget_bias=0))
# Weights and biases for output softmax layer.
out_weights = tf.Variable(tf.random_normal([self.units, self.num_class]))
out_weights = tf.Variable(tf.random.normal([self.units, self.num_class]))
out_bias = tf.Variable(tf.zeros([self.num_class]))
# Transpose input x to make it time major.