mirrors/icefall
1
0
Fork 0
mirror of https://github.com/k2-fsa/icefall.git synced 2025-12-11 06:55:27 +00:00
Code Issues Packages Projects Releases Wiki Activity

style check

Browse Source
This commit is contained in:
luomingshuang 2022-05-06 18:52:24 +08:00
parent cdb6591014
commit c3b2a522b0
5 changed files with 10 additions and 12 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
egs/wenetspeech/ASR/local/display_manifest_statistics.py
View File

@ -31,12 +31,11 @@ from lhotse import load_manifest
def main(): def main():
paths = [ paths = [
#"./data/fbank/cuts_S.jsonl.gz", "./data/fbank/cuts_S.jsonl.gz",
#"./data/fbank/cuts_M.jsonl.gz", "./data/fbank/cuts_M.jsonl.gz",
"./data/fbank/cuts_L.jsonl.gz", "./data/fbank/cuts_DEV.jsonl.gz",
#"./data/fbank/cuts_DEV.jsonl.gz", "./data/fbank/cuts_TEST_NET.jsonl.gz",
#"./data/fbank/cuts_TEST_NET.jsonl.gz", "./data/fbank/cuts_TEST_MEETING.jsonl.gz",
#"./data/fbank/cuts_TEST_MEETING.jsonl.gz"
] ]
for path in paths: for path in paths:

3
egs/wenetspeech/ASR/pruned_transducer_stateless2/beam_search.py
View File

@ -503,8 +503,7 @@ def modified_beam_search(
for i in range(batch_size): for i in range(batch_size):
topk_log_probs, topk_indexes = ragged_log_probs[i].topk(beam) topk_log_probs, topk_indexes = ragged_log_probs[i].topk(beam)
#topk_hyp_indexes = (topk_indexes // vocab_size).tolist() topk_hyp_indexes = (topk_indexes // vocab_size).tolist()
topk_hyp_indexes = torch.div(topk_indexes, vocab_size, rounding_mode="trunc")
topk_token_indexes = (topk_indexes % vocab_size).tolist() topk_token_indexes = (topk_indexes % vocab_size).tolist()
for k in range(len(topk_hyp_indexes)): for k in range(len(topk_hyp_indexes)):