From 26d642bfdf59a50365a9c8158acb223cae1004dc Mon Sep 17 00:00:00 2001
From: zhifu gao <zhifu.gzf@alibaba-inc.com>
Date: 星期二, 23 四月 2024 20:13:44 +0800
Subject: [PATCH] Dev gzf exp (#1651)
---
funasr/models/seaco_paraformer/model.py | 4 ++--
1 files changed, 2 insertions(+), 2 deletions(-)
diff --git a/funasr/models/seaco_paraformer/model.py b/funasr/models/seaco_paraformer/model.py
index 8f87340..b28de94 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
@@ -350,7 +350,7 @@
pre_acoustic_embeds, pre_token_length = predictor_outs[0], predictor_outs[1]
pre_token_length = pre_token_length.round().long()
if torch.max(pre_token_length) < 1:
- return []
+ return ([],)
decoder_out = self._seaco_decode_with_ASF(encoder_out,
encoder_out_lens,
--
Gitblit v1.9.1