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

fix typos

Browse Source
This commit is contained in:
yaozengwei 2022-05-30 22:37:47 +08:00
parent 603b707fa8
commit 7307f1c6bc
1 changed files with 4 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
egs/librispeech/ASR/conv_emformer_transducer_stateless/emformer.py
View File

@ -818,7 +818,7 @@ class EmformerEncoderLayer(nn.Module):
) -> Tuple[torch.Tensor, torch.Tensor, torch.Tensor]: ) -> Tuple[torch.Tensor, torch.Tensor, torch.Tensor]:
"""Unpack cached states including: """Unpack cached states including:
1) output memory from previous chunks in the lower layer; 1) output memory from previous chunks in the lower layer;
2) attention key and value of left context from proceeding chunk's 2) attention key and value of left context from preceding chunk's
computation. computation.
""" """
past_length = state[3][0][0].item() past_length = state[3][0][0].item()
@ -934,7 +934,7 @@ class EmformerEncoderLayer(nn.Module):
"""Apply attention module in inference mode. """Apply attention module in inference mode.
1) Unpack cached states including: 1) Unpack cached states including:
- memory from previous chunks in the lower layer; - memory from previous chunks in the lower layer;
- attention key and value of left context from proceeding - attention key and value of left context from preceding
chunk's compuation; chunk's compuation;
2) Apply attention computation; 2) Apply attention computation;
3) Pack updated states including: 3) Pack updated states including:
@ -1468,7 +1468,7 @@ class EmformerEncoder(nn.Module):
utterance frames for i-th batch element in x, which contains the utterance frames for i-th batch element in x, which contains the
right_context at the end. right_context at the end.
states (List[List[torch.Tensor]], optional): states (List[List[torch.Tensor]], optional):
Cached states from proceeding chunk's computation, where each Cached states from preceding chunk's computation, where each
element (List[torch.Tensor]) corresponds to each emformer layer. element (List[torch.Tensor]) corresponds to each emformer layer.
(default: None) (default: None)
conv_caches (List[torch.Tensor], optional): conv_caches (List[torch.Tensor], optional):
@ -1650,7 +1650,7 @@ class Emformer(EncoderInterface):
utterance frames for i-th batch element in x, containing the utterance frames for i-th batch element in x, containing the
right_context at the end. right_context at the end.
states (List[List[torch.Tensor]], optional): states (List[List[torch.Tensor]], optional):
Cached states from proceeding chunk's computation, where each Cached states from preceding chunk's computation, where each
element (List[torch.Tensor]) corresponds to each emformer layer. element (List[torch.Tensor]) corresponds to each emformer layer.
(default: None) (default: None)
conv_caches (List[torch.Tensor], optional): conv_caches (List[torch.Tensor], optional):