mirrors/icefall
1
0
Fork 0
mirror of https://github.com/k2-fsa/icefall.git synced 2025-08-16 12:42:20 +00:00
Code Issues Packages Projects Releases Wiki Activity

set default value of cr_loss_masked_scale to 1.0

Browse Source
This commit is contained in:
yaozengwei 2024-09-06 10:32:36 +08:00
parent 07d6b12364
commit cf796eefed
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
egs/librispeech/ASR/zipformer/model.py
View File

@ -189,7 +189,7 @@ class AsrModel(nn.Module):
targets: torch.Tensor, targets: torch.Tensor,
target_lengths: torch.Tensor, target_lengths: torch.Tensor,
time_mask: Optional[torch.Tensor] = None, 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]: ) -> Tuple[torch.Tensor, torch.Tensor]:
"""Compute CTC loss with consistency regularization loss. """Compute CTC loss with consistency regularization loss.
Args: Args:
@ -359,7 +359,7 @@ class AsrModel(nn.Module):
spec_augment: Optional[SpecAugment] = None, spec_augment: Optional[SpecAugment] = None,
supervision_segments: Optional[torch.Tensor] = None, supervision_segments: Optional[torch.Tensor] = None,
time_warp_factor: Optional[int] = 80, 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]: ) -> Tuple[torch.Tensor, torch.Tensor, torch.Tensor, torch.Tensor, torch.Tensor]:
""" """
Args: Args: