From 65d1005fd2cd5566ac819aa6f41e43dff9c8a691 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=E5=BF=97=E6=B5=A9?= <neo.dzh@alibaba-inc.com>
Date: Mon, 27 Feb 2023 14:46:01 +0800
Subject: [PATCH] fixbug for sd and sv

---
 funasr/tasks/diar.py | 6 ++++--
 1 file changed, 4 insertions(+), 2 deletions(-)

diff --git a/funasr/tasks/diar.py b/funasr/tasks/diar.py
index 7f154ef6c..73c51e350 100644
--- a/funasr/tasks/diar.py
+++ b/funasr/tasks/diar.py
@@ -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