shixian.shi před 2 roky
rodič
revize
c9f1b4e8a2
1 změnil soubory, kde provedl 1 přidání a 1 odebrání
  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