speech_recognition/FunASR
1
0
Fork 0
mirror of https://github.com/modelscope/FunASR synced 2025-09-15 14:48:36 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

modify unit test for speech_diarization_sond-en-us-callhome-8k-n16k4-pytorch

Browse Source
This commit is contained in:
志浩 2023-03-10 14:33:21 +08:00
parent c1f5bc2e4f
commit 773ab31780
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
funasr/models/encoder/opennmt_encoders/self_attention_encoder.py
View File

@ -272,7 +272,7 @@ class SelfAttentionEncoder(AbsEncoder):
position embedded tensor and mask
"""
masks = (~make_pad_mask(ilens)[:, None, :]).to(xs_pad.device)
xs_pad *= self.output_size()**0.5
xs_pad = xs_pad * self.output_size()**0.5
if self.embed is None:
xs_pad = xs_pad
elif (