From 7a55cac34611f3e87f6e386fdf6875537e648ee7 Mon Sep 17 00:00:00 2001 From: Daniel Povey Date: Sat, 22 Oct 2022 20:02:54 +0800 Subject: [PATCH] Increase max_factor in final lm_balancer and am_balancer --- egs/librispeech/ASR/pruned_transducer_stateless7/model.py | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) diff --git a/egs/librispeech/ASR/pruned_transducer_stateless7/model.py b/egs/librispeech/ASR/pruned_transducer_stateless7/model.py index f19bf1a93..d4d33303a 100644 --- a/egs/librispeech/ASR/pruned_transducer_stateless7/model.py +++ b/egs/librispeech/ASR/pruned_transducer_stateless7/model.py @@ -72,11 +72,13 @@ class Transducer(nn.Module): self.lm_balancer = ActivationBalancer(vocab_size, channel_dim=-1, min_positive=0, max_positive=1, - min_abs=0, max_abs=50.0) + min_abs=0, max_abs=50.0, + max_factor=0.1) self.am_balancer = ActivationBalancer(vocab_size, channel_dim=-1, min_positive=0, max_positive=1, - min_abs=0, max_abs=20.0) + min_abs=0, max_abs=30.0, + max_factor=0.1) def forward( self,