mirrors/icefall
1
0
Fork 0
mirror of https://github.com/k2-fsa/icefall.git synced 2025-12-11 06:55:27 +00:00
Code Issues Packages Projects Releases Wiki Activity

Bug-fix RE random shift

Browse Source
This commit is contained in:
Daniel Povey 2022-11-18 14:53:03 +08:00
parent f7c99ed1d1
commit 0601dd72fd
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/pruned_transducer_stateless7/zipformer.py
View File

@ -978,7 +978,7 @@ class CompactRelPositionalEncoding(torch.nn.Module):
""" """
shift = 0 if self.training else int(round(random.normalvariate(0, 1) * float(self.random_shift))) shift = int(round(random.normalvariate(0, 1) * float(self.random_shift))) if self.training else 0
self.extend_pe(x, shift) self.extend_pe(x, shift)
pos_emb = self.pe[ pos_emb = self.pe[