From a6eead6c982de726cceea8cb07a90e9bd18f2070 Mon Sep 17 00:00:00 2001 From: yaozengwei Date: Mon, 9 Sep 2024 10:10:08 +0800 Subject: [PATCH] minor fix --- egs/librispeech/ASR/zipformer/train.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/egs/librispeech/ASR/zipformer/train.py b/egs/librispeech/ASR/zipformer/train.py index c2aaf8b56..3fde55de2 100755 --- a/egs/librispeech/ASR/zipformer/train.py +++ b/egs/librispeech/ASR/zipformer/train.py @@ -747,7 +747,7 @@ def get_model(params: AttributeDict) -> nn.Module: def get_spec_augment(params: AttributeDict) -> SpecAugment: - num_frame_masks = 10 * params.time_mask_ratio + num_frame_masks = int(10 * params.time_mask_ratio) max_frames_mask_fraction = 0.15 * params.time_mask_ratio logging.info( f"num_frame_masks: {num_frame_masks}, "