From d2dc3af1a69ee4075bcfc0c83dc0fb8e3fc1db4e Mon Sep 17 00:00:00 2001
From: yhliang <68215459+yhliang-aslp@users.noreply.github.com>
Date: 星期四, 11 五月 2023 16:31:40 +0800
Subject: [PATCH] Merge pull request #492 from alibaba-damo-academy/dev_smohan
---
egs/aishell/rnnt/conf/train_conformer_rnnt_unified.yaml | 30 ++++++++++++------------------
1 files changed, 12 insertions(+), 18 deletions(-)
diff --git a/egs/aishell/rnnt/conf/train_conformer_rnnt_unified.yaml b/egs/aishell/rnnt/conf/train_conformer_rnnt_unified.yaml
index 60f796c..8a1c40c 100644
--- a/egs/aishell/rnnt/conf/train_conformer_rnnt_unified.yaml
+++ b/egs/aishell/rnnt/conf/train_conformer_rnnt_unified.yaml
@@ -1,32 +1,26 @@
+encoder: chunk_conformer
encoder_conf:
- main_conf:
- pos_wise_act_type: swish
- pos_enc_dropout_rate: 0.5
- conv_mod_act_type: swish
+ activation_type: swish
+ positional_dropout_rate: 0.5
time_reduction_factor: 2
unified_model_training: true
default_chunk_size: 16
jitter_range: 4
left_chunk_size: 0
- input_conf:
- block_type: conv2d
- conv_size: 512
+ embed_vgg_like: false
subsampling_factor: 4
- num_frame: 1
- body_conf:
- - block_type: conformer
- linear_size: 2048
- hidden_size: 512
- heads: 8
+ linear_units: 2048
+ output_size: 512
+ attention_heads: 8
dropout_rate: 0.5
- pos_wise_dropout_rate: 0.5
- att_dropout_rate: 0.5
- conv_mod_kernel_size: 15
+ positional_dropout_rate: 0.5
+ attention_dropout_rate: 0.5
+ cnn_module_kernel: 15
num_blocks: 12
# decoder related
-decoder: rnn
-decoder_conf:
+rnnt_decoder: rnnt
+rnnt_decoder_conf:
embed_size: 512
hidden_size: 512
embed_dropout_rate: 0.5
--
Gitblit v1.9.1