From e558be1cb79a042e4b3ae6354e28a1f774041c2d Mon Sep 17 00:00:00 2001 From: PingFeng Luo Date: Tue, 22 Feb 2022 14:41:50 +0800 Subject: [PATCH] just remove unnecessary torch.sum --- egs/aishell/ASR/transducer_stateless/model.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/egs/aishell/ASR/transducer_stateless/model.py b/egs/aishell/ASR/transducer_stateless/model.py index 0322edeed..c19325a15 100644 --- a/egs/aishell/ASR/transducer_stateless/model.py +++ b/egs/aishell/ASR/transducer_stateless/model.py @@ -122,4 +122,4 @@ class Transducer(nn.Module): loss = k2.rnnt_loss(logits, y_padded, blank_id, boundary) - return torch.sum(loss) + return loss