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

resolve typos and import issues

Browse Source
This commit is contained in:
Bailey Hirota 2025-07-09 14:06:29 +09:00
parent 8b25152edf
commit 6ef0fec6e8
1 changed files with 4 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
egs/multi_ja_en/ASR/local/utils/asr_datamodule.py
View File

@ -15,7 +15,7 @@
# See the License for the specific language governing permissions and # See the License for the specific language governing permissions and
# limitations under the License. # limitations under the License.
import torch
import argparse import argparse
import inspect import inspect
import logging import logging
@ -34,6 +34,7 @@ from lhotse.dataset import (
SpecAugment, SpecAugment,
) )
from lhotse.dataset.input_strategies import OnTheFlyFeatures from lhotse.dataset.input_strategies import OnTheFlyFeatures
from lhotse.utils import fix_random_seed
from torch.utils.data import DataLoader from torch.utils.data import DataLoader
from icefall.utils import str2bool from icefall.utils import str2bool
@ -202,7 +203,7 @@ class MultiDatasetAsrDataModule:
logging.info("Enable MUSAN") logging.info("Enable MUSAN")
logging.info("About to get Musan cuts") logging.info("About to get Musan cuts")
cuts_musan = load_manifest(self.args.manifest_dir / "musan_cuts.jsonl.gz") cuts_musan = load_manifest(self.args.manifest_dir / "musan_cuts.jsonl.gz")
transforms.append(CutMix(cuts=cuts_musan, p=0.5, snr=(10,20), preserve_id=True) transforms.append(CutMix(cuts=cuts_musan, p=0.5, snr=(10,20), preserve_id=True))
else: else:
logging.info("Disable MUSAN") logging.info("Disable MUSAN")
@ -214,6 +215,7 @@ class MultiDatasetAsrDataModule:
logging.info( logging.info(
f"Using cut concatenation with duration factor " f"Using cut concatenation with duration factor "
f"{self.args.duration_factor} and gap {self.args.gap}." f"{self.args.duration_factor} and gap {self.args.gap}."
)
transforms = [ transforms = [
CutConcatenate( CutConcatenate(
duration_factor=self.args.duration_factor, gap=self.args.gap) duration_factor=self.args.duration_factor, gap=self.args.gap)