This commit is contained in:
shixian.shi 2023-11-23 20:39:52 +08:00
parent e5528b586d
commit c9f1b4e8a2

View File

@ -134,7 +134,7 @@ class NeatContextualParaformer(Paraformer):
text_lengths: torch.Tensor, text_lengths: torch.Tensor,
hotword_pad: torch.Tensor, hotword_pad: torch.Tensor,
hotword_lengths: torch.Tensor, hotword_lengths: torch.Tensor,
ideal_attn: torch.Tensor, dha_pad: torch.Tensor,
) -> Tuple[torch.Tensor, Dict[str, torch.Tensor], torch.Tensor]: ) -> Tuple[torch.Tensor, Dict[str, torch.Tensor], torch.Tensor]:
"""Frontend + Encoder + Decoder + Calc loss """Frontend + Encoder + Decoder + Calc loss