Mirrors/modelscope
1
0
Fork 0
mirror of https://github.com/modelscope/modelscope.git synced 2026-09-01 19:49:03 +02:00
Code Issues Packages Projects Releases Wiki Activity

Update finetune_speech_recognition.py

Browse Source 
MsDataset replace by ASRDataset.
This commit is contained in:
fq
2023-07-11 11:54:05 +08:00
committed by GitHub
parent 20c15d3aaa
commit 2073f4fd55
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
examples/pytorch/auto_speech_recognition/finetune_speech_recognition.py
View File

@@ -10,7 +10,7 @@ def modelscope_finetune(params):
if not os.path.exists(params.output_dir):
os.makedirs(params.output_dir, exist_ok=True)
# dataset split ["train", "validation"]
ds_dict = MsDataset.load(params.data_path, namespace='speech_asr', download_mode=None)
ds_dict = ASRDataset.load(params.data_path, namespace='speech_asr', download_mode=None)
kwargs = dict(
model=params.model,
data_dir=ds_dict,