From 4091bf66c509224c8f8c1ed28f9411ce48e34ab4 Mon Sep 17 00:00:00 2001
From: 游雁 <zhifu.gzf@alibaba-inc.com>
Date: 星期四, 25 一月 2024 20:07:56 +0800
Subject: [PATCH] bugfix time_speech_total_per_sample=0

---
 funasr/models/transformer/model.py |    2 +-
 1 files changed, 1 insertions(+), 1 deletions(-)

diff --git a/funasr/models/transformer/model.py b/funasr/models/transformer/model.py
index e2367a7..7e40060 100644
--- a/funasr/models/transformer/model.py
+++ b/funasr/models/transformer/model.py
@@ -348,7 +348,7 @@
         scorers["ngram"] = ngram
         
         weights = dict(
-            decoder=1.0 - kwargs.get("decoding_ctc_weight"),
+            decoder=1.0 - kwargs.get("decoding_ctc_weight", 0.0),
             ctc=kwargs.get("decoding_ctc_weight", 0.0),
             lm=kwargs.get("lm_weight", 0.0),
             ngram=kwargs.get("ngram_weight", 0.0),

--
Gitblit v1.9.1