From c9f1aeb7d18eaa33c5d8b7f1fe7365ac9a0ff971 Mon Sep 17 00:00:00 2001 From: Daniel Povey Date: Fri, 18 Mar 2022 16:40:24 +0800 Subject: [PATCH] Fix bug with import --- egs/librispeech/ASR/pruned_transducer_stateless2/scaling.py | 1 + 1 file changed, 1 insertion(+) diff --git a/egs/librispeech/ASR/pruned_transducer_stateless2/scaling.py b/egs/librispeech/ASR/pruned_transducer_stateless2/scaling.py index d03bd0967..2d0331312 100644 --- a/egs/librispeech/ASR/pruned_transducer_stateless2/scaling.py +++ b/egs/librispeech/ASR/pruned_transducer_stateless2/scaling.py @@ -459,6 +459,7 @@ class ScaledEmbedding(nn.Module): self.weight[self.padding_idx].fill_(0) def forward(self, input: Tensor) -> Tensor: + F = torch.nn.functional scale = self.scale.exp() if input.numel() < self.num_embeddings: return F.embedding(