From f6c73924306085dadab9ae69caeff152eb8defa8 Mon Sep 17 00:00:00 2001 From: Daniel Povey Date: Thu, 18 May 2023 15:37:33 +0800 Subject: [PATCH] Bug fix --- egs/libriheavy/LM/zipformer1/subformer.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/egs/libriheavy/LM/zipformer1/subformer.py b/egs/libriheavy/LM/zipformer1/subformer.py index e8ad00501..d0776594b 100644 --- a/egs/libriheavy/LM/zipformer1/subformer.py +++ b/egs/libriheavy/LM/zipformer1/subformer.py @@ -875,7 +875,7 @@ class LearnedDownsamplingModule(nn.Module): if random.random() < 0.01 or __name__ == '__main__': logging.info(f"mean weight={weights.mean()}, mean-abs-scores={scores.abs().mean()} positive-scores={(scores>0).to(torch.float32).mean()}, discarded-weights={weights_discarded.mean()}, seq_len={seq_len}, seq_len_reduced={seq_len_reduced}") - weights_discarded = weights_discarded.flip(dims=1) + weights_discarded = weights_discarded.flip(dims=(1,)) weights = (weights[:, :seq_len_reduced] - weights_discarded) else: