diff --git a/espnet2/asr/maskctc_model.py b/espnet2/asr/maskctc_model.py index ab45c625606..26cf7a90956 100644 --- 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