mirrors/icefall
1
0
Fork 0
mirror of https://github.com/k2-fsa/icefall.git synced 2025-12-10 22:45:27 +00:00
Code Issues Packages Projects Releases Wiki Activity

Remove changes to files outside of relevant recipes

Browse Source
This commit is contained in:
Kinan Martin 2025-08-29 16:52:13 +09:00
parent 36fc1f1d1e
commit 7231cf44aa
3 changed files with 3 additions and 17 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
egs/aishell/ASR/conformer_mmi/transformer.py
View File

@ -545,7 +545,6 @@ class TransformerDecoderLayer(nn.Module):
memory_mask: Optional[torch.Tensor] = None, memory_mask: Optional[torch.Tensor] = None,
tgt_key_padding_mask: Optional[torch.Tensor] = None, tgt_key_padding_mask: Optional[torch.Tensor] = None,
memory_key_padding_mask: Optional[torch.Tensor] = None, memory_key_padding_mask: Optional[torch.Tensor] = None,
**kwargs,
) -> torch.Tensor: ) -> torch.Tensor:
"""Pass the inputs (and mask) through the decoder layer. """Pass the inputs (and mask) through the decoder layer.

1
egs/tedlium3/ASR/conformer_ctc2/transformer.py
View File

@ -612,7 +612,6 @@ class TransformerDecoderLayer(nn.Module):
tgt_key_padding_mask: Optional[torch.Tensor] = None, tgt_key_padding_mask: Optional[torch.Tensor] = None,
memory_key_padding_mask: Optional[torch.Tensor] = None, memory_key_padding_mask: Optional[torch.Tensor] = None,
warmup: float = 1.0, warmup: float = 1.0,
**kwargs,
) -> torch.Tensor: ) -> torch.Tensor:
"""Pass the inputs (and mask) through the decoder layer. """Pass the inputs (and mask) through the decoder layer.

18
icefall/utils.py
View File

@ -1391,20 +1391,13 @@ def add_eos(ragged: k2.RaggedTensor, eos_id: int) -> k2.RaggedTensor:
return concat(ragged, eos_id, direction="right") return concat(ragged, eos_id, direction="right")
def make_pad_mask( def make_pad_mask(lengths: torch.Tensor, max_len: int = 0) -> torch.Tensor:
lengths: torch.Tensor,
max_len: int = 0,
pad_left: bool = False,
) -> torch.Tensor:
""" """
Args: Args:
lengths: lengths:
A 1-D tensor containing sentence lengths. A 1-D tensor containing sentence lengths.
max_len: max_len:
The length of masks. The length of masks.
pad_left:
If ``False`` (default), padding is on the right.
If ``True``, padding is on the left.
Returns: Returns:
Return a 2-D bool tensor, where masked positions Return a 2-D bool tensor, where masked positions
are filled with `True` and non-masked positions are are filled with `True` and non-masked positions are
@ -1421,14 +1414,9 @@ def make_pad_mask(
max_len = max(max_len, lengths.max()) max_len = max(max_len, lengths.max())
n = lengths.size(0) n = lengths.size(0)
seq_range = torch.arange(0, max_len, device=lengths.device) seq_range = torch.arange(0, max_len, device=lengths.device)
expanded_lengths = seq_range.unsqueeze(0).expand(n, max_len) expaned_lengths = seq_range.unsqueeze(0).expand(n, max_len)
if pad_left: return expaned_lengths >= lengths.unsqueeze(-1)
mask = expanded_lengths < (max_len - lengths).unsqueeze(1)
else:
mask = expanded_lengths >= lengths.unsqueeze(-1)
return mask
# Copied and modified from https://github.com/wenet-e2e/wenet/blob/main/wenet/utils/mask.py # Copied and modified from https://github.com/wenet-e2e/wenet/blob/main/wenet/utils/mask.py