Skip to content

Commit c196372

Browse files
author
josh
committed
scorer
1 parent 664c548 commit c196372

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

evaluate.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -144,10 +144,10 @@ def create_windows(features):
144144
# Second pass, decode logits and compute WER and edit distance metrics
145145
for logits, batch in bar(zip(logitses, split_data(test_data, FLAGS.test_batch_size))):
146146
seq_lengths = batch['features_len'].values.astype(np.int32)
147-
# decoded = ctc_beam_search_decoder_batch(logits, seq_lengths, Config.alphabet, FLAGS.beam_width,
148-
# num_processes=num_processes, scorer=scorer)
149147
decoded = ctc_beam_search_decoder_batch(logits, seq_lengths, Config.alphabet, FLAGS.beam_width,
150-
num_processes=num_processes)
148+
num_processes=num_processes, scorer=scorer)
149+
# decoded = ctc_beam_search_decoder_batch(logits, seq_lengths, Config.alphabet, FLAGS.beam_width,
150+
# num_processes=num_processes)
151151

152152
ground_truths.extend(Config.alphabet.decode(l.astype(np.uint8)) for l in batch['transcript'])
153153
predictions.extend(d[0][1] for d in decoded)

0 commit comments

Comments
 (0)