add kwargs in forward argument

chintu619 · May 9, 2022 · 79cda74 · 79cda74
1 parent 537f9b6
commit 79cda74
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