From 8f4bfc7efd880ca53989e79bb2df7b11e126ac2e Mon Sep 17 00:00:00 2001 From: Yifan Yang <64255737+yfyeung@users.noreply.github.com> Date: Mon, 22 May 2023 16:58:10 +0800 Subject: [PATCH] Update egs/librispeech/ASR/pruned_transducer_stateless7_ctc_bs/frame_reducer.py Co-authored-by: Fangjun Kuang --- .../ASR/pruned_transducer_stateless7_ctc_bs/frame_reducer.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/egs/librispeech/ASR/pruned_transducer_stateless7_ctc_bs/frame_reducer.py b/egs/librispeech/ASR/pruned_transducer_stateless7_ctc_bs/frame_reducer.py index 9798fd4fe..d48887426 100644 --- a/egs/librispeech/ASR/pruned_transducer_stateless7_ctc_bs/frame_reducer.py +++ b/egs/librispeech/ASR/pruned_transducer_stateless7_ctc_bs/frame_reducer.py @@ -74,7 +74,7 @@ class FrameReducer(nn.Module): padding_mask = make_pad_mask(x_lens) non_blank_mask = (ctc_output[:, :, blank_id] < math.log(0.9)) * (~padding_mask) - if y_lens is not None or self.training == False: + if y_lens is not None or self.training is False: # Limit the maximum number of reduced frames if y_lens is not None: limit_lens = T - y_lens