From 1ad439f96b8cdbca98ee95b886d692dfa20e85ce Mon Sep 17 00:00:00 2001
From: mengzhe.cmz <mengzhe.cmz@alibaba-inc.com>
Date: 星期三, 12 四月 2023 19:25:29 +0800
Subject: [PATCH] fix task.py with no dest_sample_rate task; fix bug in train and infer
---
funasr/models/predictor/cif.py | 2 --
1 files changed, 0 insertions(+), 2 deletions(-)
diff --git a/funasr/models/predictor/cif.py b/funasr/models/predictor/cif.py
index dd92dba..e80a915 100644
--- a/funasr/models/predictor/cif.py
+++ b/funasr/models/predictor/cif.py
@@ -243,8 +243,6 @@
pre_alphas_length = cache["cif_alphas"].size(-1)
mask_chunk_peak_predictor[:, :pre_alphas_length] = 1.0
mask_chunk_peak_predictor[:, pre_alphas_length + cache["pad_left"]:pre_alphas_length + cache["stride"] + cache["pad_left"]] = 1.0
- #if cache["is_final"]:
- # mask_chunk_peak_predictor[:, -1] = 1.0
if mask_chunk_peak_predictor is not None:
cif_peak = cif_peak * mask_chunk_peak_predictor.squeeze(-1)
--
Gitblit v1.9.1