mirror of
https://github.com/k2-fsa/icefall.git
synced 2025-09-08 08:34:19 +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 = x_next
|
||||
|
||||
mask = (torch.randn_like(x) > 0.5)
|
||||
mask = (torch.rand_like(x) > 0.5)
|
||||
x = x - (x * mask) * 2
|
||||
|
||||
x = (x * self.rand_scales) + (x_bypass * self.nonrand_scales)
|
||||
|
Loading…
x
Reference in New Issue
Block a user