From a2aca9f64371b3be66cba65bac9f6b60346a9126 Mon Sep 17 00:00:00 2001 From: Daniel Povey Date: Wed, 30 Mar 2022 21:42:15 +0800 Subject: [PATCH] Bug-fix --- egs/librispeech/ASR/pruned_transducer_stateless2/joiner.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/egs/librispeech/ASR/pruned_transducer_stateless2/joiner.py b/egs/librispeech/ASR/pruned_transducer_stateless2/joiner.py index 973a89bfe..d76a913a5 100644 --- a/egs/librispeech/ASR/pruned_transducer_stateless2/joiner.py +++ b/egs/librispeech/ASR/pruned_transducer_stateless2/joiner.py @@ -44,4 +44,4 @@ class Joiner(nn.Module): logit = self.output_linear(torch.tanh(logit)) - return output + return logit