mirrors/icefall
1
0
Fork 0
mirror of https://github.com/k2-fsa/icefall.git synced 2025-09-03 06:04:18 +00:00
Code Issues Packages Projects Releases Wiki Activity

change for asr_datamodule.py

Browse Source
This commit is contained in:
luomingshuang 2022-03-07 16:32:52 +08:00
parent 93aaa14cda
commit b67c57de92
1 changed files with 11 additions and 11 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

22
egs/tedlium3/ASR/transducer_stateless/asr_datamodule.py
View File

@ -210,17 +210,17 @@ class TedLiumAsrDataModule:
logging.info(
f"Time warp factor: {self.args.spec_aug_time_warp_factor}"
)
# Judge num_frame_masks according to Lhotse's version
num_frame_masks = (
2
if (
inspect.signature(SpecAugment.__init__)
.parameters["num_frame_masks"]
.default
== 1
)
else 10
)
# Design 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
else:
num_frame_masks = 10
logging.info(f"Num frame mask: {num_frame_masks}")
input_transforms.append(
SpecAugment(