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

minor fixes

Browse Source
This commit is contained in:
JinZr 2023-07-25 16:05:57 +08:00
parent 90cb518398
commit 3b4fa4863f
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
egs/librispeech/ASR/zipformer_label_level_algn/model.py
View File

@ -265,7 +265,7 @@ class AsrModel(nn.Module):
# project_input=False since we applied the decoder's input projections
# prior to do_rnnt_pruning (this is an optimization for speed).
logits = self.joiner(
am_pruned, lm_pruned, encoder_out_lens, project_input=False
am_pruned, lm_pruned, encoder_out_lens, apply_attn=True, project_input=False
)
with torch.cuda.amp.autocast(enabled=False):