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

fix style check

Browse Source
This commit is contained in:
Guo Liyong 2021-11-01 19:29:33 +08:00
parent 4ae1701bba
commit bf98c0fd27
1 changed files with 2 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
egs/librispeech/ASR/conformer_ctc/gigaspeech_datamodule.py
View File

@ -26,11 +26,9 @@ from lhotse import CutSet, Fbank, FbankConfig, load_manifest
from lhotse.dataset import ( from lhotse.dataset import (
BucketingSampler, BucketingSampler,
CutConcatenate, CutConcatenate,
CutMix,
K2SpeechRecognitionDataset, K2SpeechRecognitionDataset,
PrecomputedFeatures, PrecomputedFeatures,
SingleCutSampler, SingleCutSampler,
SpecAugment,
) )
from lhotse.dataset.input_strategies import OnTheFlyFeatures from lhotse.dataset.input_strategies import OnTheFlyFeatures
from torch.utils.data import DataLoader from torch.utils.data import DataLoader
@ -195,7 +193,6 @@ class GigaSpeechAsrDataModule(DataModule):
input_strategy=OnTheFlyFeatures( input_strategy=OnTheFlyFeatures(
Fbank(FbankConfig(num_mel_bins=80)) Fbank(FbankConfig(num_mel_bins=80))
), ),
input_transforms=input_transforms,
return_cuts=self.args.return_cuts, return_cuts=self.args.return_cuts,
) )
@ -321,15 +318,11 @@ class GigaSpeechAsrDataModule(DataModule):
@lru_cache() @lru_cache()
def valid_cuts(self) -> CutSet: def valid_cuts(self) -> CutSet:
logging.info("About to get dev cuts") logging.info("About to get dev cuts")
cuts_valid = load_manifest( cuts_valid = load_manifest(self.args.feature_dir / "cuts_DEV.json.gz")
self.args.feature_dir / "cuts_DEV.json.gz"
)
return cuts_valid return cuts_valid
@lru_cache() @lru_cache()
def test_cuts(self) -> List[CutSet]: def test_cuts(self) -> List[CutSet]:
logging.info("About to get dev cuts") logging.info("About to get dev cuts")
cuts_test = load_manifest( cuts_test = load_manifest(self.args.feature_dir / "cuts_TEST.json.gz")
self.args.feature_dir / "cuts_TEST.json.gz"
)
return cuts_test return cuts_test