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)
|
initial_scale=0.05)
|
||||||
|
|
||||||
self.whiten = Whiten(num_groups=1,
|
self.whiten = Whiten(num_groups=1,
|
||||||
whitening_limit=_whitening_schedule(7.5),
|
whitening_limit=5.0,
|
||||||
prob=(0.025, 0.25),
|
prob=(0.025, 0.25),
|
||||||
grad_scale=0.01)
|
grad_scale=0.01)
|
||||||
|
|
||||||
|
|||||||
Loading…
x
Reference in New Issue
Block a user