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

fixed feat dim

Browse Source
This commit is contained in:
zr_jin 2023-07-23 18:20:55 +08:00
parent bb8d016722
commit 3bd2e8e6cc
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/alignment_attention_module.py
View File

@ -336,7 +336,7 @@ class AlignmentAttentionModule(nn.Module):
attn_weights = self.cross_attn_weights(lm_pruned, am_pruned, pos_emb)
label_level_am_representation = self.cross_attn(am_pruned, attn_weights)
return label_level_am_representation
return label_level_am_representation.reshape(batch_size, T, prune_range, encoder_dim)
if __name__ == "__main__":