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

Fix transformer decoder layer

Browse Source
This commit is contained in:
k2-fsa 2025-07-18 20:08:31 +08:00
parent 9fd0f2dc1d
commit e2b29afd1d
8 changed files with 8 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -545,6 +545,7 @@ 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/aishell/ASR/conformer_mmi/transformer.py
View File

@ -545,6 +545,7 @@ 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/gigaspeech/ASR/conformer_ctc/transformer.py
View File

@ -549,6 +549,7 @@ 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/librispeech/ASR/conformer_ctc/transformer.py
View File

@ -549,6 +549,7 @@ 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/librispeech/ASR/conformer_ctc2/transformer.py
View File

@ -550,6 +550,7 @@ 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.

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

@ -537,6 +537,7 @@ 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/librispeech/ASR/streaming_conformer_ctc/transformer.py
View File

@ -567,6 +567,7 @@ 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,6 +612,7 @@ 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.