mirror of
https://github.com/k2-fsa/icefall.git
synced 2025-12-11 06:55:27 +00:00
Change whitening limit of NonlinAttentionModule from _whitening_schedule(7.5), to 5.0
This commit is contained in:
parent
025bcc155d
commit
dcf6fced40
@ -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)
|
||||
|
||||
|
||||
Loading…
x
Reference in New Issue
Block a user