Reduce initial whitening-schedule limit from 7.5 to 5.0 for NonlinAttentionModule.

This commit is contained in:
Daniel Povey 2022-12-03 14:40:39 +08:00
parent b8e3091e04
commit 306fd85bab

View File

@ -1488,7 +1488,7 @@ class NonlinAttentionModule(nn.Module):
initial_scale=0.05)
self.whiten = Whiten(num_groups=1,
whitening_limit=_whitening_schedule(7.5),
whitening_limit=_whitening_schedule(5.0),
prob=(0.025, 0.25),
grad_scale=0.01)