From c2dee5e3c29eba79e591d9e9caebaef15ea4e56b Mon Sep 17 00:00:00 2001
From: hnluo <haoneng.lhn@alibaba-inc.com>
Date: 星期四, 29 六月 2023 11:09:28 +0800
Subject: [PATCH] Merge pull request #687 from alibaba-damo-academy/dev_lhn
---
egs/aishell/rnnt/conf/train_conformer_rnnt_unified.yaml | 35 +++++++++++++++++++++++++++--------
1 files changed, 27 insertions(+), 8 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..a1f27a3 100644
--- a/egs/aishell/rnnt/conf/train_conformer_rnnt_unified.yaml
+++ b/egs/aishell/rnnt/conf/train_conformer_rnnt_unified.yaml
@@ -6,7 +6,7 @@
unified_model_training: true
default_chunk_size: 16
jitter_range: 4
- left_chunk_size: 0
+ left_chunk_size: 1
embed_vgg_like: false
subsampling_factor: 4
linear_units: 2048
@@ -25,24 +25,33 @@
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
grad_clip: 5
-max_epoch: 200
+max_epoch: 120
val_scheduler_criterion:
- valid
- loss
@@ -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