mirror of
https://github.com/k2-fsa/icefall.git
synced 2025-12-11 06:55:27 +00:00
Merge branch 'scaled_adam_exp535' into scaled_adam_exp548
This commit is contained in:
commit
f48786534a
@ -1414,8 +1414,8 @@ class FeedforwardModule(nn.Module):
|
||||
self.hidden_balancer = ActivationBalancer(feedforward_dim,
|
||||
channel_dim=-1,
|
||||
min_positive=ScheduledFloat((0.0, 0.1), (8000.0, 0.05)),
|
||||
min_abs=1.0,
|
||||
max_abs=ScheduledFloat((0.0, 5.0), (8000.0, 10.0), default=10),
|
||||
min_abs=1.5,
|
||||
max_abs=15.0,
|
||||
min_prob=0.25)
|
||||
self.activation = DoubleSwish()
|
||||
self.dropout = nn.Dropout(dropout)
|
||||
|
||||
Loading…
x
Reference in New Issue
Block a user