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

Reduce initial_speed

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

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

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