From 25c16704310c933ddf766cd05f3c45d15f125c53 Mon Sep 17 00:00:00 2001
From: yifanyeung <yifanyeung@yifanyeung.local>
Date: Fri, 19 Jan 2024 17:48:00 +0800
Subject: [PATCH] update

---
 egs/tal_csasr/ASR/pruned_transducer_stateless5/asr_datamodule.py | 1 -
 .../ASR/pruned_transducer_stateless2/asr_datamodule.py           | 1 -
 2 files changed, 2 deletions(-)

diff --git a/egs/tal_csasr/ASR/pruned_transducer_stateless5/asr_datamodule.py b/egs/tal_csasr/ASR/pruned_transducer_stateless5/asr_datamodule.py
index 43e61036e..6f0833db6 100644
--- a/egs/tal_csasr/ASR/pruned_transducer_stateless5/asr_datamodule.py
+++ b/egs/tal_csasr/ASR/pruned_transducer_stateless5/asr_datamodule.py
@@ -309,7 +309,6 @@ class TAL_CSASRAsrDataModule:
                 buffer_size=self.args.num_buckets * 2000,
                 shuffle_buffer_size=self.args.num_buckets * 5000,
                 num_cuts_for_bins_estimate=20000,
-                buffer_size=60000,
                 drop_last=self.args.drop_last,
             )
         else:
diff --git a/egs/wenetspeech/ASR/pruned_transducer_stateless2/asr_datamodule.py b/egs/wenetspeech/ASR/pruned_transducer_stateless2/asr_datamodule.py
index c39aff41d..58da1d68c 100644
--- a/egs/wenetspeech/ASR/pruned_transducer_stateless2/asr_datamodule.py
+++ b/egs/wenetspeech/ASR/pruned_transducer_stateless2/asr_datamodule.py
@@ -294,7 +294,6 @@ class WenetSpeechAsrDataModule:
                 num_buckets=self.args.num_buckets,
                 buffer_size=self.args.num_buckets * 2000,
                 shuffle_buffer_size=self.args.num_buckets * 5000,
-                buffer_size=300000,
                 drop_last=True,
             )
         else: