From 705b70b495c2ac40b281180823e71239fb6127b7 Mon Sep 17 00:00:00 2001 From: JRMeyer Date: Wed, 12 Dec 2018 05:27:58 +0300 Subject: [PATCH] ctc_loss longer than input --- evaluate.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/evaluate.py b/evaluate.py index fd6b45ec..417e1f04 100755 --- a/evaluate.py +++ b/evaluate.py @@ -117,7 +117,8 @@ def evaluate(test_data, inference_graph, alphabet): sparse_labels = tf.cast(ctc_label_dense_to_sparse(labels_ph, label_lengths_ph, FLAGS.test_batch_size), tf.int32) loss = tf.nn.ctc_loss(labels=sparse_labels, inputs=layers['raw_logits'], - sequence_length=inputs['input_lengths']) + sequence_length=inputs['input_lengths'], + ignore_longer_outputs_than_inputs=True) # Create a saver using variables from the above newly created graph mapping = {v.op.name: v for v in tf.global_variables() if not v.op.name.startswith('previous_state_')}