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

fixbug for sd and sv

Browse Source
This commit is contained in:
志浩 2023-02-27 14:46:01 +08:00
parent 3b42ace3d4
commit 65d1005fd2
1 changed files with 4 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
funasr/tasks/diar.py
View File

@ -124,9 +124,9 @@ encoder_choices = ClassChoices(
resnet34=ResNet34Diar,
sanm_chunk_opt=SANMEncoderChunkOpt,
data2vec_encoder=Data2VecEncoder,
epaca_dtnn=ECAPA_TDNN,
ecapa_tdnn=ECAPA_TDNN,
),
type_check=AbsEncoder,
type_check=torch.nn.Module,
default="resnet34",
)
speaker_encoder_choices = ClassChoices(
@ -189,6 +189,8 @@ class DiarTask(AbsTask):
specaug_choices,
# --normalize and --normalize_conf
normalize_choices,
# --label_aggregator and --label_aggregator_conf
label_aggregator_choices,
# --model and --model_conf
model_choices,
# --encoder and --encoder_conf