mirrors/icefall
1
0
Fork 0
mirror of https://github.com/k2-fsa/icefall.git synced 2025-08-14 12:32:20 +00:00
Code Issues Packages Projects Releases Wiki Activity

update comments

Browse Source
This commit is contained in:
glynpu 2023-03-17 12:17:08 +08:00
parent c055f0cc49
commit 0a5b639ec1
1 changed files with 6 additions and 14 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

20
egs/himia/wuw/ctc_tdnn/asr_datamodule.py
View File

@ -16,7 +16,6 @@
import argparse
import inspect
import logging
from functools import lru_cache
from pathlib import Path
@ -190,7 +189,7 @@ class HiMiaWuwDataModule:
"--input-strategy",
type=str,
default="PrecomputedFeatures",
help="PrecomputedFeatures",
help="AudioSamples or PrecomputedFeatures",
)
group.add_argument(
"--train-channel",
@ -198,6 +197,8 @@ class HiMiaWuwDataModule:
default="_7_01",
help="""channel of HI_MIA train dataset.
All channels are used if it is set "all".
Please refer state 6 in prepare.sh for its meaning and other
potential values. Currently, Only "_7_01" is verified.
""",
)
group.add_argument(
@ -206,6 +207,8 @@ class HiMiaWuwDataModule:
default="_7_01",
help="""channel of HI_MIA dev dataset.
All channels are used if it is set "all".
Please refer state 6 in prepare.sh for its meaning and other
potential values. Currently, Only "_7_01" is verified.
""",
)
@ -248,22 +251,11 @@ class HiMiaWuwDataModule:
input_transforms = []
if self.args.enable_spec_aug:
logging.info("Enable SpecAugment")
logging.info(f"Time warp factor: {self.args.spec_aug_time_warp_factor}")
# Set the value of num_frame_masks according to Lhotse's version.
# In different Lhotse's versions, the default of num_frame_masks is
# different.
num_frame_masks = 10
num_frame_masks_parameter = inspect.signature(
SpecAugment.__init__
).parameters["num_frame_masks"]
if num_frame_masks_parameter.default == 1:
num_frame_masks = 2
logging.info(f"Num frame mask: {num_frame_masks}")
input_transforms.append(
SpecAugment(
time_warp_factor=self.args.spec_aug_time_warp_factor,
num_frame_masks=num_frame_masks,
num_frame_masks=10,
features_mask_size=27,
num_feature_masks=2,
frames_mask_size=100,