From e59db01b7c599afb0c780a33289b4b86c6579afe Mon Sep 17 00:00:00 2001 From: Daniel Povey Date: Thu, 31 Mar 2022 13:03:26 +0800 Subject: [PATCH] Reduce initial_speed --- egs/librispeech/ASR/pruned_transducer_stateless2/model.py | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) diff --git a/egs/librispeech/ASR/pruned_transducer_stateless2/model.py b/egs/librispeech/ASR/pruned_transducer_stateless2/model.py index ab729a429..0355c4531 100644 --- a/egs/librispeech/ASR/pruned_transducer_stateless2/model.py +++ b/egs/librispeech/ASR/pruned_transducer_stateless2/model.py @@ -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