experiments/STT
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

In ctc_beam_search_decoder(), added a sanity check between input class_dim and alphabet

Browse Source
This commit is contained in:
godeffroy 2020-08-24 08:30:12 +02:00
parent a54b198d1e
commit 95b6fccaf1
1 changed files with 1 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
native_client/ctcdecode/ctc_beam_search_decoder.cpp
View File

@ -226,6 +226,7 @@ std::vector<Output> ctc_beam_search_decoder(
std::shared_ptr<Scorer> ext_scorer,
size_t num_results)
{
VALID_CHECK_EQ(alphabet.GetSize()+1, class_dim, "Number of output classes in acoustic model does not match number of labels in the alphabet file. Alphabet file must be the same one that was used to train the acoustic model.");
DecoderState state;
state.init(alphabet, beam_size, cutoff_prob, cutoff_top_n, ext_scorer);
state.next(probs, time_dim, class_dim);