mirror of
https://github.com/k2-fsa/icefall.git
synced 2025-09-10 09:34:39 +00:00
Bug fix
This commit is contained in:
parent
a83bde1372
commit
e7886d49a9
@ -900,7 +900,7 @@ class Decorrelate(torch.nn.Module):
|
|||||||
x_next.scatter_(-1, perm, x)
|
x_next.scatter_(-1, perm, x)
|
||||||
x = x_next
|
x = x_next
|
||||||
|
|
||||||
mask = (torch.randn_like(x) > 0.5)
|
mask = (torch.rand_like(x) > 0.5)
|
||||||
x = x - (x * mask) * 2
|
x = x - (x * mask) * 2
|
||||||
|
|
||||||
x = (x * self.rand_scales) + (x_bypass * self.nonrand_scales)
|
x = (x * self.rand_scales) + (x_bypass * self.nonrand_scales)
|
||||||
|
Loading…
x
Reference in New Issue
Block a user