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

add doc

Browse Source
This commit is contained in:
yaozengwei 2022-06-26 22:34:40 +08:00
parent 7b15596495
commit 5ea58a4465
1 changed files with 2 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
egs/librispeech/ASR/conv_emformer_transducer_stateless3/emformer.py
View File

@ -2109,6 +2109,8 @@ class RelPositionalEncoding(torch.nn.Module):
self.pe_negative = self.pe_negative.to(dtype=dtype, device=device) self.pe_negative = self.pe_negative.to(dtype=dtype, device=device)
pe = torch.cat( pe = torch.cat(
[ [
# it starts from the min negative value of relative position
# and it is bound to be gathered
self.pe_negative[self.neg_len - neg_len :], self.pe_negative[self.neg_len - neg_len :],
self.pe_positive[1:pos_len], self.pe_positive[1:pos_len],
], ],