mirrors/icefall
1
0
Fork 0
mirror of https://github.com/k2-fsa/icefall.git synced 2025-08-09 10:02:22 +00:00
Code Issues Packages Projects Releases Wiki Activity

Fix bug with import

Browse Source
This commit is contained in:
Daniel Povey 2022-03-18 16:40:24 +08:00
parent 2dfcd8f117
commit c9f1aeb7d1
1 changed files with 1 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
egs/librispeech/ASR/pruned_transducer_stateless2/scaling.py
View File

@ -459,6 +459,7 @@ class ScaledEmbedding(nn.Module):
self.weight[self.padding_idx].fill_(0) self.weight[self.padding_idx].fill_(0)
def forward(self, input: Tensor) -> Tensor: def forward(self, input: Tensor) -> Tensor:
F = torch.nn.functional
scale = self.scale.exp() scale = self.scale.exp()
if input.numel() < self.num_embeddings: if input.numel() < self.num_embeddings:
return F.embedding( return F.embedding(