From 30d0bc6ad7d3e2fa0cc9f3b210dafeca6e953e37 Mon Sep 17 00:00:00 2001 From: Daniel Povey Date: Sun, 27 Nov 2022 17:17:11 +0800 Subject: [PATCH] Make gain factor 4 times larger, for constraining the sign in ActivationBalancer. --- egs/librispeech/ASR/pruned_transducer_stateless7/scaling.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/egs/librispeech/ASR/pruned_transducer_stateless7/scaling.py b/egs/librispeech/ASR/pruned_transducer_stateless7/scaling.py index eb31755e1..ebe20a2d1 100644 --- a/egs/librispeech/ASR/pruned_transducer_stateless7/scaling.py +++ b/egs/librispeech/ASR/pruned_transducer_stateless7/scaling.py @@ -604,7 +604,7 @@ class ActivationBalancer(torch.nn.Module): min_positive: FloatLike = 0.05, max_positive: FloatLike = 0.95, max_factor: FloatLike = 0.04, - sign_gain_factor: FloatLike = 0.01, + sign_gain_factor: FloatLike = 1.0, scale_gain_factor: FloatLike = 0.02, min_abs: FloatLike = 0.2, max_abs: FloatLike = 100.0,