From c49f1eead78036adc2fb23c5711f28aa2563de45 Mon Sep 17 00:00:00 2001 From: PingFeng Luo Date: Wed, 19 Jan 2022 17:29:21 +0800 Subject: [PATCH] fix style --- egs/wenetspeech/ASR/transducer_stateless/model.py | 9 ++++++--- 1 file changed, 6 insertions(+), 3 deletions(-) diff --git a/egs/wenetspeech/ASR/transducer_stateless/model.py b/egs/wenetspeech/ASR/transducer_stateless/model.py index 46050e5ab..9f9455bdd 100644 --- a/egs/wenetspeech/ASR/transducer_stateless/model.py +++ b/egs/wenetspeech/ASR/transducer_stateless/model.py @@ -107,13 +107,16 @@ class Transducer(nn.Module): max_sym_id = torch.max(y_padded) assert encoder_out.size(0) == decoder_out.size(0), [ encoder_out.size(), - decoder_out.size()] + decoder_out.size(), + ] assert encoder_out.size(2) == decoder_out.size(2), [ encoder_out.size(), - decoder_out.size()] + decoder_out.size(), + ] assert encoder_out.size(2) >= (max_sym_id + 1), [ encoder_out.size(), - max_sym_id] + max_sym_id, + ] boundary = torch.zeros( (x.size(0), 4), dtype=torch.int64, device=x.device