From cc2c1d1d53dea5d2c45f858d1baa5bd279f47987 Mon Sep 17 00:00:00 2001
From: nichongjia-2007 <nichongjia@gmail.com>
Date: 星期三, 31 五月 2023 14:39:25 +0800
Subject: [PATCH] Merge branch 'main' of https://github.com/alibaba-damo-academy/FunASR

---
 egs/aishell/rnnt/conf/train_conformer_rnnt_unified.yaml |   31 +++++++++++++++++++++++++------
 1 files changed, 25 insertions(+), 6 deletions(-)

diff --git a/egs/aishell/rnnt/conf/train_conformer_rnnt_unified.yaml b/egs/aishell/rnnt/conf/train_conformer_rnnt_unified.yaml
index 8a1c40c..59f9936 100644
--- a/egs/aishell/rnnt/conf/train_conformer_rnnt_unified.yaml
+++ b/egs/aishell/rnnt/conf/train_conformer_rnnt_unified.yaml
@@ -25,19 +25,28 @@
     hidden_size: 512
     embed_dropout_rate: 0.5
     dropout_rate: 0.5
-
 joint_network_conf:
     joint_space_size: 512
 
+# frontend related
+frontend: wav_frontend
+frontend_conf:
+    fs: 16000
+    window: hamming
+    n_mels: 80
+    frame_length: 25
+    frame_shift: 10
+    lfr_m: 1
+    lfr_n: 1
+
+
 # Auxiliary CTC
+model: rnnt_unified
 model_conf:
     auxiliary_ctc_weight: 0.0
 
 # minibatch related
 use_amp: true
-batch_type: unsorted
-batch_size: 16
-num_workers: 16
 
 # optimization related
 accum_grad: 1
@@ -59,8 +68,6 @@
 scheduler_conf:
    warmup_steps: 25000
 
-normalize: None
-
 specaug: specaug
 specaug_conf:
     apply_time_warp: true
@@ -77,4 +84,16 @@
     - 50
     num_time_mask: 5
 
+dataset_conf:
+    data_names: speech,text
+    data_types: sound,text
+    shuffle: True
+    shuffle_conf:
+        shuffle_size: 2048
+        sort_size: 500
+    batch_conf:
+        batch_type: token
+        batch_size: 16000
+    num_workers: 8
+
 log_interval: 50

--
Gitblit v1.9.1