Skip to content

Commit

Permalink
Update trainer
Browse files Browse the repository at this point in the history
  • Loading branch information
sooftware committed Feb 3, 2021
1 parent cff5a8a commit ad36c04
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion kospeech/trainer/supervised_trainer.py
Original file line number Diff line number Diff line change
Expand Up @@ -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:
Expand Down

0 comments on commit ad36c04

Please sign in to comment.