mirrors/icefall
1
0
Fork 0
mirror of https://github.com/k2-fsa/icefall.git synced 2025-09-03 14:14:19 +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( logging.info(
f"Time warp factor: {self.args.spec_aug_time_warp_factor}" f"Time warp factor: {self.args.spec_aug_time_warp_factor}"
) )
# Judge num_frame_masks according to Lhotse's version # Design the value of num_frame_masks according to Lhotse's version.
num_frame_masks = ( # In different Lhotse's versions, the default of num_frame_masks is
2 # different.
if ( num_frame_masks = 10
inspect.signature(SpecAugment.__init__) num_frame_masks_parameter = inspect.signature(
.parameters["num_frame_masks"] SpecAugment.__init__
.default ).parameters["num_frame_masks"]
== 1 if num_frame_masks_parameter.default == 1:
) num_frame_masks = 2
else 10 else:
) num_frame_masks = 10
logging.info(f"Num frame mask: {num_frame_masks}") logging.info(f"Num frame mask: {num_frame_masks}")
input_transforms.append( input_transforms.append(
SpecAugment( SpecAugment(