From 79cda74ba20f0b795251e23a9cb9fd624e2be02d Mon Sep 17 00:00:00 2001 From: Yosuke Higuchi Date: Mon, 9 May 2022 22:43:29 +0900 Subject: [PATCH] add kwargs in forward argument --- espnet2/asr/maskctc_model.py | 1 + 1 file changed, 1 insertion(+) 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