From 3a4281f4959534b1bf5d01acf0085f4f8e6f2ec8 Mon Sep 17 00:00:00 2001
From: wuhongsheng <664116298@qq.com>
Date: 星期五, 05 七月 2024 00:55:32 +0800
Subject: [PATCH] 优化speakid和语句匹配逻辑，部分解决speakid不从0递增问题 (#1870)
---
funasr/download/download_dataset_from_hub.py | 19 +++++++++++--------
1 files changed, 11 insertions(+), 8 deletions(-)
diff --git a/funasr/download/download_dataset_from_hub.py b/funasr/download/download_dataset_from_hub.py
index d06d821..2419ce8 100644
--- a/funasr/download/download_dataset_from_hub.py
+++ b/funasr/download/download_dataset_from_hub.py
@@ -1,11 +1,14 @@
-
def download_dataset():
- pass
+ pass
+
def download_dataset_from_ms(**kwargs):
- from modelscope.msdatasets import MsDataset
- dataset_name = kwargs.get("dataset_name", 'speech_asr/speech_asr_aishell1_trainsets')
- subset_name = kwargs.get("subset_name", 'default')
- split = kwargs.get("split", 'train')
- data_dump_dir = kwargs.get("data_dump_dir", None)
- ds = MsDataset.load(dataset_name=dataset_name, subset_name=subset_name, split=split, cache_dir=data_dump_dir)
\ No newline at end of file
+ from modelscope.msdatasets import MsDataset
+
+ dataset_name = kwargs.get("dataset_name", "speech_asr/speech_asr_aishell1_trainsets")
+ subset_name = kwargs.get("subset_name", "default")
+ split = kwargs.get("split", "train")
+ data_dump_dir = kwargs.get("data_dump_dir", None)
+ ds = MsDataset.load(
+ dataset_name=dataset_name, subset_name=subset_name, split=split, cache_dir=data_dump_dir
+ )
--
Gitblit v1.9.1