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

Minor fixes after review.

Browse Source
This commit is contained in:
Fangjun Kuang 2022-01-25 18:46:35 +08:00
parent dd2acd89fd
commit 4749619e5a
1 changed files with 2 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
egs/librispeech/ASR/transducer_stateless/conformer.py
View File

@ -785,13 +785,9 @@ class RelPositionMultiheadAttention(nn.Module):
matrix_ac + matrix_bd
) * scaling # (batch, head, time1, time2)
attn_output_weights = attn_output_weights.view(
bsz * num_heads, tgt_len, -1
)
# attn_output_weights is of shape (bsz, num_heads, tgt_len, src_len)
attn_output_weights = attn_output_weights.view(
bsz, num_heads, tgt_len, -1
).permute(0, 2, 3, 1)
attn_output_weights = attn_output_weights.permute(0, 2, 3, 1)
# now attn_output_weights is of shape (bsz, tgt_len, src_len, num_heads)
attn_output_weights = torch.matmul(