shixian.shi 2 ani în urmă
părinte
comite
c9f1b4e8a2
1 a modificat fișierele cu 1 adăugiri și 1 ștergeri
  1. 1 1
      funasr/models/e2e_asr_contextual_paraformer.py

+ 1 - 1
funasr/models/e2e_asr_contextual_paraformer.py

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