From 02f2a3c2ec07bfedb10cf6906c3df413fc8258f1 Mon Sep 17 00:00:00 2001
From: speech_asr <wangjiaming.wjm@alibaba-inc.com>
Date: 星期四, 20 四月 2023 11:38:20 +0800
Subject: [PATCH] update
---
funasr/utils/build_model.py | 6 ++++--
1 files changed, 4 insertions(+), 2 deletions(-)
diff --git a/funasr/utils/build_model.py b/funasr/utils/build_model.py
index b7646a2..0384de6 100644
--- a/funasr/utils/build_model.py
+++ b/funasr/utils/build_model.py
@@ -217,7 +217,7 @@
encoder_class = encoder_choices.get_class(args.encoder)
encoder = encoder_class(input_size=input_size, **args.encoder_conf)
- # 7. Decoder
+ # decoder
decoder_class = decoder_choices.get_class(args.decoder)
decoder = decoder_class(
vocab_size=vocab_size,
@@ -225,7 +225,9 @@
**args.decoder_conf,
)
- # 8. CTC
+ # ctc
ctc = CTC(
odim=vocab_size, encoder_output_size=encoder.output_size(), **args.ctc_conf
)
+
+
--
Gitblit v1.9.1