Fix typo exporting the wrong tensor for TFLite model
This commit is contained in:
parent
1c71263790
commit
d6642da05b
@ -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 (
|
||||
{
|
||||
|
Loading…
x
Reference in New Issue
Block a user