维石
2024-04-18 3d12aa567443d603941636ab6308b4aafc5a80b0
update seaco finetune
1个文件已修改
2 ■■■ 已修改文件
funasr/models/seaco_paraformer/model.py 2 ●●● 补丁 | 查看 | 原始文档 | blame | 历史
funasr/models/seaco_paraformer/model.py
@@ -157,7 +157,7 @@
                                        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