From a4a920cc22f6e07699bd7aca349ce6f4d95ec88c Mon Sep 17 00:00:00 2001 From: PingFeng Luo Date: Wed, 19 Jan 2022 17:20:28 +0800 Subject: [PATCH] fix style --- egs/wenetspeech/ASR/transducer_stateless/model.py | 12 ++++++------ 1 file changed, 6 insertions(+), 6 deletions(-) diff --git a/egs/wenetspeech/ASR/transducer_stateless/model.py b/egs/wenetspeech/ASR/transducer_stateless/model.py index 016c58594..c324b57bf 100644 --- a/egs/wenetspeech/ASR/transducer_stateless/model.py +++ b/egs/wenetspeech/ASR/transducer_stateless/model.py @@ -105,12 +105,12 @@ class Transducer(nn.Module): y_padded = y_padded.to(torch.int64) max_sym_id = torch.max(y_padded) - assert encoder_out.size(0) == decoder_out.size(0),\ - [encoder_out.size(), decoder_out.size()] - assert encoder_out.size(2) == decoder_out.size(2),\ - [encoder_out.size(), decoder_out.size()] - assert encoder_out.size(2) >= (max_sym_id + 1),\ - [encoder_out.size(), max_sym_id] + assert encoder_out.size(0) == decoder_out.size(0), [ + encoder_out.size(), decoder_out.size()] + assert encoder_out.size(2) == decoder_out.size(2), [ + encoder_out.size(), decoder_out.size()] + assert encoder_out.size(2) >= (max_sym_id + 1), [ + encoder_out.size(), max_sym_id] boundary = torch.zeros( (x.size(0), 4), dtype=torch.int64, device=x.device