From d6642da05baa2a461ef0943d1ccaa2cdfd91c7ba Mon Sep 17 00:00:00 2001 From: Alexandre Lissy Date: Wed, 7 Nov 2018 16:52:57 +0100 Subject: [PATCH] Fix typo exporting the wrong tensor for TFLite model --- DeepSpeech.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/DeepSpeech.py b/DeepSpeech.py index ea9503da..58749722 100755 --- a/DeepSpeech.py +++ b/DeepSpeech.py @@ -1828,7 +1828,7 @@ def create_inference_graph(batch_size=1, n_steps=16, use_new_decoder=False, tfli else: logits = tf.identity(logits, name='logits') new_state_c = tf.identity(new_state_c, name='new_state_c') - new_state_h = tf.identity(new_state_c, name='new_state_h') + new_state_h = tf.identity(new_state_h, name='new_state_h') return ( {