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

update conv_emformer_transducer/emformer.py.

Browse Source
This commit is contained in:
yaozengwei 2022-04-10 20:29:22 +08:00
parent 8129470586
commit a24eef8096
1 changed files with 2 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
egs/librispeech/ASR/conv_emformer_transducer/emformer.py
View File

@ -92,6 +92,8 @@ class EmformerAttention(nn.Module):
self.emb_to_query = nn.Linear(embed_dim, embed_dim, bias=True) self.emb_to_query = nn.Linear(embed_dim, embed_dim, bias=True)
self.out_proj = nn.Linear(embed_dim, embed_dim, bias=True) self.out_proj = nn.Linear(embed_dim, embed_dim, bias=True)
self._reset_parameters()
def _reset_parameters(self) -> None: def _reset_parameters(self) -> None:
nn.init.xavier_uniform_(self.emb_to_key_value.weight) nn.init.xavier_uniform_(self.emb_to_key_value.weight)
nn.init.constant_(self.emb_to_key_value.bias, 0.0) nn.init.constant_(self.emb_to_key_value.bias, 0.0)