Update asr_datamodule.py

This commit is contained in:
jinzr 2024-09-08 19:04:37 +08:00
parent cea0dbe7b1
commit fecee6d12c

View File

@ -82,7 +82,7 @@ class AlimeetingAsrDataModule:
group.add_argument( group.add_argument(
"--manifest-dir", "--manifest-dir",
type=Path, type=Path,
default=Path("data/manifests"), default=Path("data/fbank"),
help="Path to directory with train/valid/test cuts.", help="Path to directory with train/valid/test cuts.",
) )
group.add_argument( group.add_argument(
@ -327,9 +327,11 @@ class AlimeetingAsrDataModule:
def test_dataloaders(self, cuts: CutSet) -> DataLoader: def test_dataloaders(self, cuts: CutSet) -> DataLoader:
logging.debug("About to create test dataset") logging.debug("About to create test dataset")
test = K2SpeechRecognitionDataset( test = K2SpeechRecognitionDataset(
input_strategy=OnTheFlyFeatures(Fbank(FbankConfig(num_mel_bins=80))) input_strategy=(
if self.args.on_the_fly_feats OnTheFlyFeatures(Fbank(FbankConfig(num_mel_bins=80)))
else PrecomputedFeatures(), if self.args.on_the_fly_feats
else PrecomputedFeatures()
),
return_cuts=True, return_cuts=True,
) )
sampler = DynamicBucketingSampler( sampler = DynamicBucketingSampler(