From 7828c6ff7325e37f4afab94a65f6d049f3d3e05d Mon Sep 17 00:00:00 2001 From: Fangjun Kuang Date: Thu, 30 Dec 2021 00:15:23 +0800 Subject: [PATCH] Minor fixes. --- egs/librispeech/ASR/transducer_stateless/joiner.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/egs/librispeech/ASR/transducer_stateless/joiner.py b/egs/librispeech/ASR/transducer_stateless/joiner.py index 6ee22deec..9fd9da4f1 100644 --- a/egs/librispeech/ASR/transducer_stateless/joiner.py +++ b/egs/librispeech/ASR/transducer_stateless/joiner.py @@ -65,8 +65,8 @@ class Joiner(nn.Module): x = [p.reshape(-1, self.input_dim) for p in x] x = torch.cat(x) - logit = torch.tanh(x) + activations = torch.tanh(x) - output = self.output_linear(logit) + logits = self.output_linear(activations) - return output + return logits