mirrors/icefall
1
0
Fork 0
mirror of https://github.com/k2-fsa/icefall.git synced 2025-08-09 10:02:22 +00:00
Code Issues Packages Projects Releases Wiki Activity

Update asr_datamodule.py (#1619)

Browse Source
This commit is contained in:
zr_jin 2024-05-07 21:37:55 +08:00 committed by GitHub
parent 4e97b19b63
commit 9d570870cf
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 6 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
egs/yesno/ASR/tdnn/asr_datamodule.py
View File

@ -181,7 +181,7 @@ class YesNoAsrDataModule(DataModule):
train = K2SpeechRecognitionDataset(
cut_transforms=transforms,
input_strategy=OnTheFlyFeatures(
FbankConfig(sampling_rate=8000, num_mel_bins=23)
Fbank(FbankConfig(sampling_rate=8000, num_mel_bins=23))
),
return_cuts=self.args.return_cuts,
)
@ -222,9 +222,11 @@ class YesNoAsrDataModule(DataModule):
logging.debug("About to create test dataset")
test = K2SpeechRecognitionDataset(
input_strategy=OnTheFlyFeatures(Fbank(FbankConfig(num_mel_bins=23)))
if self.args.on_the_fly_feats
else PrecomputedFeatures(),
input_strategy=(
OnTheFlyFeatures(Fbank(FbankConfig(num_mel_bins=23)))
if self.args.on_the_fly_feats
else PrecomputedFeatures()
),
return_cuts=self.args.return_cuts,
)
sampler = DynamicBucketingSampler(