update seaco finetune

This commit is contained in:
维石 2024-04-18 10:14:16 +08:00
parent e8f80e96f9
commit 3d12aa5674

View File

@ -157,7 +157,7 @@ class SeacoParaformer(BiCifParaformer, Paraformer):
seaco_label_pad,
)
if self.train_decoder:
loss_att, acc_att = self._calc_att_loss(
loss_att, acc_att, _, _, _ = self._calc_att_loss(
encoder_out, encoder_out_lens, text, text_lengths
)
loss = loss_seaco + loss_att * self.seaco_weight