From 187a302be238d1e6a757871f77e42117ff221c64 Mon Sep 17 00:00:00 2001
From: shixian.shi <shixian.shi@alibaba-inc.com>
Date: 星期二, 27 六月 2023 16:11:26 +0800
Subject: [PATCH] update clas finetune
---
funasr/datasets/large_datasets/utils/tokenize.py | 3 +++
1 files changed, 3 insertions(+), 0 deletions(-)
diff --git a/funasr/datasets/large_datasets/utils/tokenize.py b/funasr/datasets/large_datasets/utils/tokenize.py
index a7eb6d2..3128a06 100644
--- a/funasr/datasets/large_datasets/utils/tokenize.py
+++ b/funasr/datasets/large_datasets/utils/tokenize.py
@@ -54,6 +54,9 @@
length = len(text)
if 'hw_tag' in data:
+ if hw_config['pre_hwlist'] is not None and hw_config['pre_prob'] > 0:
+ # enable preset hotword detect in sampling
+ import pdb; pdb.set_trace()
hotword_indxs = sample_hotword(length, **hw_config)
data['hotword_indxs'] = hotword_indxs
del data['hw_tag']
--
Gitblit v1.9.1