Revert "Change whitening limit of NonlinAttentionModule from _whitening_schedule(7.5), to 5.0"

This reverts commit dcf6fced40a831892e7bb4e52eddf26e6e070eef.
This commit is contained in:
Daniel Povey 2022-12-03 00:19:48 +08:00
parent bd1b1dd7e3
commit 3faa4ffa3f

View File

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