Merge branch 'scaled_adam_exp535' into scaled_adam_exp548

This commit is contained in:
Daniel Povey 2022-11-29 15:43:44 +08:00
commit f48786534a

View File

@ -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)