From cf796eefed40fe211b22b4381a4166915a76f11f Mon Sep 17 00:00:00 2001 From: yaozengwei Date: Fri, 6 Sep 2024 10:32:36 +0800 Subject: [PATCH] set default value of cr_loss_masked_scale to 1.0 --- egs/librispeech/ASR/zipformer/model.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/egs/librispeech/ASR/zipformer/model.py b/egs/librispeech/ASR/zipformer/model.py index cf935d835..2de1e08fe 100644 --- a/egs/librispeech/ASR/zipformer/model.py +++ b/egs/librispeech/ASR/zipformer/model.py @@ -189,7 +189,7 @@ class AsrModel(nn.Module): targets: torch.Tensor, target_lengths: torch.Tensor, time_mask: Optional[torch.Tensor] = None, - cr_loss_masked_scale: float = 3.0, + cr_loss_masked_scale: float = 1.0, ) -> Tuple[torch.Tensor, torch.Tensor]: """Compute CTC loss with consistency regularization loss. Args: @@ -359,7 +359,7 @@ class AsrModel(nn.Module): spec_augment: Optional[SpecAugment] = None, supervision_segments: Optional[torch.Tensor] = None, time_warp_factor: Optional[int] = 80, - cr_loss_masked_scale: float = 3.0, + cr_loss_masked_scale: float = 1.0, ) -> Tuple[torch.Tensor, torch.Tensor, torch.Tensor, torch.Tensor, torch.Tensor]: """ Args: