diff --git a/kospeech/trainer/supervised_trainer.py b/kospeech/trainer/supervised_trainer.py
index be2e6ba7..78d99a2c 100644
--- a/kospeech/trainer/supervised_trainer.py
+++ b/kospeech/trainer/supervised_trainer.py
@@ -430,7 +430,7 @@ def _model_forward(
                         outputs, targets[:, 1:].contiguous().int(), input_lengths.int(), target_lengths.int()
                     )
                 else:
-                    outputs, output_lengths = model(inputs, input_lengths)
+                    outputs, output_lengths = model(inputs, input_lengths, targets, target_lengths)
                     loss = self.criterion(outputs.transpose(0, 1), targets[:, 1:], output_lengths, target_lengths)
             else:
                 if model.decoder is not None: