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

Reduce initial_speed from 0.5 to 0.25

Browse Source
This commit is contained in:
Daniel Povey 2022-03-31 13:47:28 +08:00
parent e59db01b7c
commit fcb0dba2cf
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
egs/librispeech/ASR/pruned_transducer_stateless2/model.py
View File

@ -62,9 +62,9 @@ class Transducer(nn.Module):
self.joiner = joiner self.joiner = joiner
self.simple_am_proj = ScaledLinear(embedding_dim, vocab_size, self.simple_am_proj = ScaledLinear(embedding_dim, vocab_size,
initial_speed=0.5) initial_speed=0.25)
self.simple_lm_proj = ScaledLinear(embedding_dim, vocab_size, self.simple_lm_proj = ScaledLinear(embedding_dim, vocab_size,
initial_speed=0.5) initial_speed=0.25)
with torch.no_grad(): with torch.no_grad():
# Initialize the two projections to be the same; this will be # Initialize the two projections to be the same; this will be
# convenient for the real joiner, which adds the endcoder # convenient for the real joiner, which adds the endcoder