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

This commit is contained in:
Daniel Povey 2022-12-01 14:28:56 +08:00
parent 025bcc155d
commit dcf6fced40

View File

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