From 362a6de3c850079ed2ae6f94181f2a51c0596527 Mon Sep 17 00:00:00 2001
From: 游雁 <zhifu.gzf@alibaba-inc.com>
Date: 星期二, 20 二月 2024 15:22:13 +0800
Subject: [PATCH] update

---
 examples/aishell/e_branchformer/conf/e_branchformer_12e_6d_2048_256.yaml |    9 ++++++---
 1 files changed, 6 insertions(+), 3 deletions(-)

diff --git a/examples/aishell/e_branchformer/conf/e_branchformer_12e_6d_2048_256.yaml b/examples/aishell/e_branchformer/conf/e_branchformer_12e_6d_2048_256.yaml
index dea580e..e6cc1e6 100644
--- a/examples/aishell/e_branchformer/conf/e_branchformer_12e_6d_2048_256.yaml
+++ b/examples/aishell/e_branchformer/conf/e_branchformer_12e_6d_2048_256.yaml
@@ -94,13 +94,16 @@
 dataset: AudioDataset
 dataset_conf:
     index_ds: IndexDSJsonl
-    batch_sampler: DynamicBatchLocalShuffleSampler
+    batch_sampler: RankFullLocalShuffleBatchSampler
     batch_type: example # example or length
-    batch_size: 1 # if batch_type is example, batch_size is the numbers of samples; if length, batch_size is source_token_len+target_token_len;
+    batch_size: 32 # if batch_type is example, batch_size is the numbers of samples; if length, batch_size is source_token_len+target_token_len;
     max_token_length: 2048 # filter samples if source_token_len+target_token_len > max_token_length,
-    buffer_size: 500
+    buffer_size: 1024
     shuffle: True
     num_workers: 4
+    preprocessor_speech: SpeechPreprocessSpeedPerturb
+    preprocessor_speech_conf:
+      speed_perturb: [0.9, 1.0, 1.1]
 
 tokenizer: CharTokenizer
 tokenizer_conf:

--
Gitblit v1.9.1