Merge pull request espnet#4347 from YosukeHiguchi/espnet2_maskctc2

Minor fix for Mask-CTC forward function
chintu619 · May 9, 2022 · beb3360 · beb3360
2 parents 537f9b6 + 79cda74
commit beb3360
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/espnet2/asr/maskctc_model.py b/espnet2/asr/maskctc_model.py
@@ -122,6 +122,7 @@ def forward(
         speech_lengths: torch.Tensor,
         text: torch.Tensor,
         text_lengths: torch.Tensor,
+        **kwargs,
     ) -> Tuple[torch.Tensor, Dict[str, torch.Tensor], torch.Tensor]:
         """Frontend + Encoder + Decoder + Calc loss