Deprecated tf.random_normal to tf.random.normal in lstm README
This commit is contained in:
parent
7a193942cd
commit
c5f4fe9b95
@ -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.
|
||||
|
Loading…
x
Reference in New Issue
Block a user