From e7886d49a9ce917ae54d2ea211207399d8dfb10c Mon Sep 17 00:00:00 2001 From: Daniel Povey Date: Wed, 8 Jun 2022 11:05:29 +0800 Subject: [PATCH] Bug fix --- egs/librispeech/ASR/pruned_transducer_stateless2/scaling.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/egs/librispeech/ASR/pruned_transducer_stateless2/scaling.py b/egs/librispeech/ASR/pruned_transducer_stateless2/scaling.py index 9015f3f3b..6e4884df7 100644 --- a/egs/librispeech/ASR/pruned_transducer_stateless2/scaling.py +++ b/egs/librispeech/ASR/pruned_transducer_stateless2/scaling.py @@ -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)