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

integrating remarks from PR

Browse Source
This commit is contained in:
Karel Vesely 2025-09-11 15:49:31 +02:00
parent c3acfcfa6c
commit ce7ec4f3cb
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
egs/librispeech/ASR/zipformer/ctc_align.py
View File

@ -270,7 +270,7 @@ def align_one_batch(
# https://github.com/pytorch/audio/blob/main/src/libtorchaudio/forced_align/gpu/compute.cu#L277
for ii in range(batch_size):
labels, log_probs = forced_align(
log_probs=ctc_output[ii, : encoder_out_lens[ii]].unsqueeze(dim=0),
log_probs=ctc_output[ii:ii+1, : encoder_out_lens[ii]],
targets=targets[ii, : target_lengths[ii]].unsqueeze(dim=0),
input_lengths=encoder_out_lens[ii].unsqueeze(dim=0),
target_lengths=target_lengths[ii].unsqueeze(dim=0),
@ -370,7 +370,7 @@ def align_dataset(
except TypeError:
num_batches = "?"
ignored_tokens = params.ignored_tokens + ["<sos/eos>", "<unk>"]
ignored_tokens = set(params.ignored_tokens + ["<sos/eos>", "<unk>"])
ignored_tokens_ints = [sp.piece_to_id(token) for token in ignored_tokens]
logging.info(f"ignored tokens {ignored_tokens}")