From 69ccdd35cda4c8482e189fa350fbcb83997872f2 Mon Sep 17 00:00:00 2001
From: wanchen.swc <wanchen.swc@alibaba-inc.com>
Date: 星期一, 06 三月 2023 18:18:31 +0800
Subject: [PATCH] [Quantization] model quantization for inference
---
funasr/tasks/diar.py | 6 ++++--
1 files changed, 4 insertions(+), 2 deletions(-)
diff --git a/funasr/tasks/diar.py b/funasr/tasks/diar.py
index 7f154ef..73c51e3 100644
--- a/funasr/tasks/diar.py
+++ b/funasr/tasks/diar.py
@@ -124,9 +124,9 @@
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 @@
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
--
Gitblit v1.9.1