From 267dddcdbb8a9200ebda5a673905d0bb512de80e Mon Sep 17 00:00:00 2001
From: 志浩 <neo.dzh@alibaba-inc.com>
Date: 星期一, 27 二月 2023 11:48:09 +0800
Subject: [PATCH] fixbug for sd and sv
---
funasr/tasks/diar.py | 2 ++
1 files changed, 2 insertions(+), 0 deletions(-)
diff --git a/funasr/tasks/diar.py b/funasr/tasks/diar.py
index 9a43945..7f154ef 100644
--- a/funasr/tasks/diar.py
+++ b/funasr/tasks/diar.py
@@ -24,6 +24,7 @@
from funasr.layers.label_aggregation import LabelAggregate
from funasr.models.ctc import CTC
from funasr.models.encoder.resnet34_encoder import ResNet34Diar
+from funasr.models.encoder.ecapa_tdnn_encoder import ECAPA_TDNN
from funasr.models.encoder.opennmt_encoders.conv_encoder import ConvEncoder
from funasr.models.encoder.opennmt_encoders.fsmn_encoder import FsmnEncoder
from funasr.models.encoder.opennmt_encoders.self_attention_encoder import SelfAttentionEncoder
@@ -123,6 +124,7 @@
resnet34=ResNet34Diar,
sanm_chunk_opt=SANMEncoderChunkOpt,
data2vec_encoder=Data2VecEncoder,
+ epaca_dtnn=ECAPA_TDNN,
),
type_check=AbsEncoder,
default="resnet34",
--
Gitblit v1.9.1