mirror of
https://github.com/k2-fsa/icefall.git
synced 2025-12-11 06:55:27 +00:00
Revert min_abs in NonlinAttentionModule from 2.0 to 1.5
This commit is contained in:
parent
f48786534a
commit
73e420865c
@ -1464,7 +1464,7 @@ class NonlinAttentionModule(nn.Module):
|
|||||||
channels // (2 * ratio), channel_dim=-1,
|
channels // (2 * ratio), channel_dim=-1,
|
||||||
min_positive=ScheduledFloat((0.0, 0.1), (8000.0, 0.05)),
|
min_positive=ScheduledFloat((0.0, 0.1), (8000.0, 0.05)),
|
||||||
max_positive=1.0,
|
max_positive=1.0,
|
||||||
min_abs=2.0,
|
min_abs=1.5,
|
||||||
max_abs=ScheduledFloat((0.0, 5.0), (8000.0, 10.0), default=1.0),
|
max_abs=ScheduledFloat((0.0, 5.0), (8000.0, 10.0), default=1.0),
|
||||||
)
|
)
|
||||||
self.sigmoid = nn.Sigmoid()
|
self.sigmoid = nn.Sigmoid()
|
||||||
|
|||||||
Loading…
x
Reference in New Issue
Block a user