update

k2-fsa · Jan 19, 2024 · 25c1670 · 25c1670
1 parent c2e769f
commit 25c1670
Show file tree

Hide file tree

Showing 2 changed files with 0 additions and 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
@@ -309,7 +309,6 @@ def train_dataloaders(
                 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
@@ -294,7 +294,6 @@ def train_dataloaders(
                 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: