From 17eaf419c05853a4ecb8dfd3a0e8ebf26a1dfb1b Mon Sep 17 00:00:00 2001
From: 嘉渊 <wangjiaming.wjm@alibaba-inc.com>
Date: 星期四, 18 五月 2023 14:10:11 +0800
Subject: [PATCH] Merge branch 'dev_infer' of https://github.com/alibaba/FunASR into dev_infer

---
 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