Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Add fast_beam_search_nbest. #420

Merged
merged 10 commits into from
Jun 21, 2022
+1,299 −195
Merged

Add fast_beam_search_nbest. #420

Changes from 1 commit
Commits
Show all changes
10 commits
Select commit Hold shift + click to select a range
1bf2e17
Add fast_beam_search_nbest.
csukuangfj Jun 14, 2022
df6ab85
Fix CI errors.
csukuangfj Jun 15, 2022
d3af7ee
Fix CI errors.
csukuangfj Jun 15, 2022
46430a5
More fixes.
csukuangfj Jun 15, 2022
feb9fd0
Small fixes.
csukuangfj Jun 15, 2022
ec9e4cf
Support using log_add in LG decoding with fast_beam_search.
csukuangfj Jun 21, 2022
136ee53
Support LG decoding in pruned_transducer_stateless
csukuangfj Jun 21, 2022
f5af662
Support LG for pruned_transducer_stateless2.
csukuangfj Jun 21, 2022
284cbf7
Support LG for fast beam search.
csukuangfj Jun 21, 2022
e032b03
Minor fixes.
csukuangfj Jun 21, 2022
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
Prev Previous commit
Next Next commit
Fix CI errors.
@csukuangfj
csukuangfj committed Jun 15, 2022
commit df6ab85fb4332b54b2217336c18acbf4be853ee4
6 changes: 6 additions & 0 deletions .github/scripts/run-librispeech-pruned-transducer-stateless5-2022-05-13.sh
View file
Original file line number Diff line number Diff line change
@@ -32,6 +32,12 @@ for sym in 1 2 3; do
--max-sym-per-frame $sym \
--checkpoint $repo/exp/pretrained.pt \
--bpe-model $repo/data/lang_bpe_500/bpe.model \
--num-encoder-layers 18 \
--dim-feedforward 2048 \
--nhead 8 \
--encoder-dim 512 \
--decoder-dim 512 \
--joiner-dim 512
$repo/test_wavs/1089-134686-0001.wav \
$repo/test_wavs/1221-135766-0001.wav \
$repo/test_wavs/1221-135766-0002.wav