From 39ce60bb7c61363414c75a6c5411959e255dadfa Mon Sep 17 00:00:00 2001 From: Daniel Povey Date: Mon, 28 Nov 2022 10:45:53 +0800 Subject: [PATCH] Decrease final value of max_abs in AttentionSqueeze from 5.0 to 1.0 --- egs/librispeech/ASR/pruned_transducer_stateless7/zipformer.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/egs/librispeech/ASR/pruned_transducer_stateless7/zipformer.py b/egs/librispeech/ASR/pruned_transducer_stateless7/zipformer.py index 91e58613a..cf78ab105 100644 --- a/egs/librispeech/ASR/pruned_transducer_stateless7/zipformer.py +++ b/egs/librispeech/ASR/pruned_transducer_stateless7/zipformer.py @@ -1307,7 +1307,7 @@ class AttentionSqueeze(nn.Module): bottleneck_dim, channel_dim=-1, min_positive=0.05, max_positive=0.95, min_abs=0.05, - max_abs=ScheduledFloat((0.0, 0.4), (20000.0, 5.0), default=1.0), + max_abs=ScheduledFloat((0.0, 0.5), (4000.0, 1.0), default=1.0), max_factor=0.02, min_prob=0.1, )