speech_recognition/FunASR
1
0
Fork 0
mirror of https://github.com/modelscope/FunASR synced 2025-09-15 14:48:36 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

update seaco finetune

Browse Source
This commit is contained in:
维石 2024-04-18 10:14:16 +08:00
parent e8f80e96f9
commit 3d12aa5674
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
funasr/models/seaco_paraformer/model.py
View File

@ -157,7 +157,7 @@ class SeacoParaformer(BiCifParaformer, Paraformer):
seaco_label_pad, seaco_label_pad,
) )
if self.train_decoder: 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 encoder_out, encoder_out_lens, text, text_lengths
) )
loss = loss_seaco + loss_att * self.seaco_weight loss = loss_seaco + loss_att * self.seaco_weight