From 8e6c7ef3e28ee1b95161ba280872c2eea3ae1fe4 Mon Sep 17 00:00:00 2001
From: Daniel Povey <dpovey@gmail.com>
Date: Fri, 16 Dec 2022 15:08:46 +0800
Subject: [PATCH] Adjust default prob of 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 b126bd99d..be66d00f6 100644
--- a/egs/librispeech/ASR/pruned_transducer_stateless7/scaling.py
+++ b/egs/librispeech/ASR/pruned_transducer_stateless7/scaling.py
@@ -547,7 +547,7 @@ class ActivationBalancer(torch.nn.Module):
 
 
         if prob is None:
-            prob = ScheduledFloat((0.0, 0.4), (8000.0, 0.1))
+            prob = ScheduledFloat((0.0, 0.4), (8000.0, 0.1), default=0.4)
         self.prob = prob
 
         # actually self.num_channels is no longer needed except for an assertion.