From e3e8b1990c2ab89b62f3a2f1d4fd57722b65336c Mon Sep 17 00:00:00 2001 From: yaozengwei Date: Fri, 8 Jul 2022 17:16:10 +0800 Subject: [PATCH] use i&i-1 to judge if it is a power of 2 --- .../ASR/conv_emformer_transducer_stateless/emformer.py | 4 ++-- .../ASR/conv_emformer_transducer_stateless2/emformer.py | 4 ++-- 2 files changed, 4 insertions(+), 4 deletions(-) diff --git a/egs/librispeech/ASR/conv_emformer_transducer_stateless/emformer.py b/egs/librispeech/ASR/conv_emformer_transducer_stateless/emformer.py index c425fe67c..fa908ffcd 100644 --- a/egs/librispeech/ASR/conv_emformer_transducer_stateless/emformer.py +++ b/egs/librispeech/ASR/conv_emformer_transducer_stateless/emformer.py @@ -1252,8 +1252,8 @@ class EmformerEncoder(nn.Module): ): super().__init__() - assert int(math.log(chunk_length, 2)) == math.log( - chunk_length, 2 + assert ( + chunk_length - 1 & chunk_length == 0 ), "chunk_length should be a power of 2." self.use_memory = memory_size > 0 diff --git a/egs/librispeech/ASR/conv_emformer_transducer_stateless2/emformer.py b/egs/librispeech/ASR/conv_emformer_transducer_stateless2/emformer.py index c1175d395..18d2047bd 100644 --- a/egs/librispeech/ASR/conv_emformer_transducer_stateless2/emformer.py +++ b/egs/librispeech/ASR/conv_emformer_transducer_stateless2/emformer.py @@ -1188,8 +1188,8 @@ class EmformerEncoder(nn.Module): ): super().__init__() - assert int(math.log(chunk_length, 2)) == math.log( - chunk_length, 2 + assert ( + chunk_length - 1 & chunk_length == 0 ), "chunk_length should be a power of 2." self.use_memory = memory_size > 0