Revert "Change whitening limit of NonlinAttentionModule from _whitening_schedule(7.5), to 5.0"
This reverts commit dcf6fced40a831892e7bb4e52eddf26e6e070eef.
This commit is contained in:
parent
bd1b1dd7e3
commit
3faa4ffa3f
@ -1488,7 +1488,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=5.0,
|
whitening_limit=_whitening_schedule(7.5),
|
||||||
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