From 3d12aa567443d603941636ab6308b4aafc5a80b0 Mon Sep 17 00:00:00 2001
From: 维石 <shixian.shi@alibaba-inc.com>
Date: 星期四, 18 四月 2024 10:14:16 +0800
Subject: [PATCH] update seaco finetune
---
funasr/models/seaco_paraformer/model.py | 2 +-
1 files changed, 1 insertions(+), 1 deletions(-)
diff --git a/funasr/models/seaco_paraformer/model.py b/funasr/models/seaco_paraformer/model.py
index 8f87340..4778a7e 100644
--- a/funasr/models/seaco_paraformer/model.py
+++ b/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
--
Gitblit v1.9.1