From 79211633ed79766466a2410c2bed8e33bab8cd8d Mon Sep 17 00:00:00 2001 From: Guanbo Wang Date: Wed, 6 Apr 2022 19:05:45 -0400 Subject: [PATCH] Use BucketingSampler for valid and test dataloader --- egs/gigaspeech/ASR/conformer_ctc/asr_datamodule.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/egs/gigaspeech/ASR/conformer_ctc/asr_datamodule.py b/egs/gigaspeech/ASR/conformer_ctc/asr_datamodule.py index 7faea33f5..9fc3d1328 100644 --- a/egs/gigaspeech/ASR/conformer_ctc/asr_datamodule.py +++ b/egs/gigaspeech/ASR/conformer_ctc/asr_datamodule.py @@ -22,6 +22,7 @@ from pathlib import Path from lhotse import CutSet, Fbank, FbankConfig, load_manifest from lhotse.dataset import ( + BucketingSampler, CutConcatenate, CutMix, DynamicBucketingSampler, @@ -320,7 +321,7 @@ class GigaSpeechAsrDataModule: cut_transforms=transforms, return_cuts=self.args.return_cuts, ) - valid_sampler = DynamicBucketingSampler( + valid_sampler = BucketingSampler( cuts_valid, max_duration=self.args.max_duration, shuffle=False, @@ -344,7 +345,7 @@ class GigaSpeechAsrDataModule: else PrecomputedFeatures(), return_cuts=self.args.return_cuts, ) - sampler = DynamicBucketingSampler( + sampler = BucketingSampler( cuts, max_duration=self.args.max_duration, shuffle=False ) logging.debug("About to create test dataloader")