mirrors/icefall
1
0
Fork 0
mirror of https://github.com/k2-fsa/icefall.git synced 2025-12-11 06:55:27 +00:00
Code Issues Packages Projects Releases Wiki Activity

Adjust min_positive and max_positive of NonlinAttentionModule

Browse Source
This commit is contained in:
Daniel Povey 2022-12-08 20:31:37 +08:00
parent e77a12602c
commit dea177fa63
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
egs/librispeech/ASR/pruned_transducer_stateless7/zipformer.py
View File

@ -1486,8 +1486,8 @@ class NonlinAttentionModule(nn.Module):
# before the sigmoid. # before the sigmoid.
self.balancer1 = ActivationBalancer( self.balancer1 = ActivationBalancer(
hidden_channels // ratio, channel_dim=-1, hidden_channels // ratio, channel_dim=-1,
min_positive=ScheduledFloat((0.0, 0.1), (8000.0, 0.05)), min_positive=ScheduledFloat((0.0, 0.25), (20000.0, 0.05)),
max_positive=1.0, max_positive=ScheduledFloat((0.0, 0.75), (20000.0, 0.95)),
min_abs=0.75, min_abs=0.75,
max_abs=ScheduledFloat((0.0, 2.5), (8000.0, 5.0), default=1.0), max_abs=ScheduledFloat((0.0, 2.5), (8000.0, 5.0), default=1.0),
) )