mirror of
https://github.com/modelscope/FunASR
synced 2025-09-15 14:48:36 +08:00
update seaco finetune
This commit is contained in:
parent
e8f80e96f9
commit
3d12aa5674
@ -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
|
||||
|
||||
Loading…
Reference in New Issue
Block a user