diff --git a/egs/librispeech/ASR/pruned_transducer_stateless7/zipformer.py b/egs/librispeech/ASR/pruned_transducer_stateless7/zipformer.py index fc83bc0e9..fe51e57ad 100644 --- a/egs/librispeech/ASR/pruned_transducer_stateless7/zipformer.py +++ b/egs/librispeech/ASR/pruned_transducer_stateless7/zipformer.py @@ -1306,7 +1306,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)), + max_abs=ScheduledFloat((0.0, 0.4), (20000.0, 5.0), default=1.0), max_factor=0.02, min_prob=0.1, )