mirrors/icefall
1
0
Fork 0
mirror of https://github.com/k2-fsa/icefall.git synced 2025-09-05 15:14: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:46:26 +08:00
parent b67c57de92
commit f1cf817cf4
1 changed files with 1 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -210,7 +210,7 @@ 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}"
) )
# Design the value of num_frame_masks according to Lhotse's version. # Set the value of num_frame_masks according to Lhotse's version.
# In different Lhotse's versions, the default of num_frame_masks is # In different Lhotse's versions, the default of num_frame_masks is
# different. # different.
num_frame_masks = 10 num_frame_masks = 10
@ -219,8 +219,6 @@ class TedLiumAsrDataModule:
).parameters["num_frame_masks"] ).parameters["num_frame_masks"]
if num_frame_masks_parameter.default == 1: if num_frame_masks_parameter.default == 1:
num_frame_masks = 2 num_frame_masks = 2
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(