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

Replace nn.Linear with ScaledLinear in simple joiner

Browse Source
This commit is contained in:
Daniel Povey 2022-03-31 12:18:31 +08:00
parent 9a0c2e7fee
commit f75d40c725
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_stateless2/model.py
View File

@ -63,7 +63,7 @@ class Transducer(nn.Module):
# could perhaps separate this into 2 linear projections, one # could perhaps separate this into 2 linear projections, one
# for lm and one for am. # for lm and one for am.
self.simple_joiner = nn.Linear(embedding_dim, vocab_size) self.simple_joiner = ScaledLinear(embedding_dim, vocab_size)
def forward( def forward(
self, self,