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-09 17:17:28 +08:00
parent 7907c3df07
commit 77cbfde968
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
funasr/models/encoder/opennmt_encoders/conv_encoder.py
View File

@ -137,12 +137,12 @@ class ConvEncoder(AbsEncoder):
self.out_padding = nn.ConstantPad1d((left_padding, right_padding), 0.0)
self.conv_out = nn.Conv1d(
num_units,
num_units,
out_units,
kernel_size,
)
if self.out_norm:
self.after_norm = LayerNorm(num_units)
self.after_norm = LayerNorm(out_units)
def output_size(self) -> int:
return self.num_units