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

Fix a bug introduced while supporting torch script. (#79)

Browse Source
This commit is contained in:
Fangjun Kuang 2021-10-14 20:09:38 +08:00 committed by GitHub
parent 5016ee3c95
commit f2387fe523
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
egs/librispeech/ASR/conformer_ctc/transformer.py
View File

@ -660,7 +660,7 @@ class PositionalEncoding(nn.Module):
self.xscale = math.sqrt(self.d_model) self.xscale = math.sqrt(self.d_model)
self.dropout = nn.Dropout(p=dropout) self.dropout = nn.Dropout(p=dropout)
# not doing: self.pe = None because of errors thrown by torchscript # not doing: self.pe = None because of errors thrown by torchscript
self.pe = torch.zeros(0, self.d_model, dtype=torch.float32) self.pe = torch.zeros(0, 0, dtype=torch.float32)
def extend_pe(self, x: torch.Tensor) -> None: def extend_pe(self, x: torch.Tensor) -> None:
"""Extend the time t in the positional encoding if required. """Extend the time t in the positional encoding if required.