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