yuekai
/

aishell1_tlg_essentials

Model card Files Files and versions

Yuekai Zhang commited on May 30, 2023

Commit

c92992e

·

1 Parent(s): 57bf40b

fix bug

Files changed (1) hide show

test/test_riva_wfst_decoder.py +1 -1

test/test_riva_wfst_decoder.py CHANGED Viewed

@@ -103,7 +103,7 @@ if __name__ == "__main__":
     ctc_log_probs = ctc_log_probs.contiguous().cuda()
     frame_reducer = FrameReducer()
-    ctc_log_probs, encoder_out_len = frame_reducer(ctc_log_probs, encoder_out_lens.cuda(), ctc_log_probs)
     vocab_size = ctc_log_probs.shape[2]
     riva_decoder = RivaWFSTDecoder(vocab_size, lang_dir, beam_size)

     ctc_log_probs = ctc_log_probs.contiguous().cuda()
     frame_reducer = FrameReducer()
+    ctc_log_probs, encoder_out_lens = frame_reducer(ctc_log_probs, encoder_out_lens.cuda(), ctc_log_probs)
     vocab_size = ctc_log_probs.shape[2]
     riva_decoder = RivaWFSTDecoder(vocab_size, lang_dir, beam_size)