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

Bug fix RE masking

Browse Source
This commit is contained in:
Daniel Povey 2022-11-09 13:12:34 +08:00
parent 20e6d2a157
commit cba194aa26
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/pruned_transducer_stateless7/zipformer.py
View File

@ -1225,7 +1225,7 @@ class RelPositionMultiheadAttentionWeights(nn.Module):
num_heads, batch_size, seq_len, seq_len
)
attn_scores = attn_scores.masked_fill(
key_padding_mask.unsqueeze(-1),
key_padding_mask.unsqueeze(1),
float("-inf"),
)