From c78474c97907c0a24c302a3952f3389360909fd9 Mon Sep 17 00:00:00 2001 From: yufan-aslp Date: Thu, 16 Feb 2023 16:31:33 +0800 Subject: [PATCH] NPU-ASLP --- .../RESULTS.md | 2 +- .../finetune.py | 4 ++-- .../speech_mfcca_asr-zh-cn-16k-alimeeting-vocab4950/infer.py | 4 ++-- .../infer_after_finetune.py | 2 +- 4 files changed, 6 insertions(+), 6 deletions(-) diff --git a/egs_modelscope/asr/mfcca/speech_mfcca_asr-zh-cn-16k-alimeeting-vocab4950/RESULTS.md b/egs_modelscope/asr/mfcca/speech_mfcca_asr-zh-cn-16k-alimeeting-vocab4950/RESULTS.md index 8f5825966..716d44eeb 100644 --- a/egs_modelscope/asr/mfcca/speech_mfcca_asr-zh-cn-16k-alimeeting-vocab4950/RESULTS.md +++ b/egs_modelscope/asr/mfcca/speech_mfcca_asr-zh-cn-16k-alimeeting-vocab4950/RESULTS.md @@ -1,5 +1,5 @@ # Paraformer-Large -- Model link: +- Model link: - Model size: 45M # Environments diff --git a/egs_modelscope/asr/mfcca/speech_mfcca_asr-zh-cn-16k-alimeeting-vocab4950/finetune.py b/egs_modelscope/asr/mfcca/speech_mfcca_asr-zh-cn-16k-alimeeting-vocab4950/finetune.py index 281292ffa..bf8176ed1 100755 --- a/egs_modelscope/asr/mfcca/speech_mfcca_asr-zh-cn-16k-alimeeting-vocab4950/finetune.py +++ b/egs_modelscope/asr/mfcca/speech_mfcca_asr-zh-cn-16k-alimeeting-vocab4950/finetune.py @@ -24,12 +24,12 @@ def modelscope_finetune(params): if __name__ == '__main__': - params = modelscope_args(model="yufan6/speech_mfcca_asr-zh-cn-16k-alimeeting-vocab4950") + params = modelscope_args(model="NPU-ASLP/speech_mfcca_asr-zh-cn-16k-alimeeting-vocab4950") params.output_dir = "./checkpoint" # m模型保存路径 params.data_path = "./example_data/" # 数据路径 params.dataset_type = "small" # 小数据量设置small,若数据量大于1000小时,请使用large params.batch_bins = 1000 # batch size,如果dataset_type="small",batch_bins单位为fbank特征帧数,如果dataset_type="large",batch_bins单位为毫秒, params.max_epoch = 10 # 最大训练轮数 params.lr = 0.0001 # 设置学习率 - params.model_revision = 'v2.0.0' + params.model_revision = 'v1.0.0' modelscope_finetune(params) diff --git a/egs_modelscope/asr/mfcca/speech_mfcca_asr-zh-cn-16k-alimeeting-vocab4950/infer.py b/egs_modelscope/asr/mfcca/speech_mfcca_asr-zh-cn-16k-alimeeting-vocab4950/infer.py index 3054394d3..fa22aad05 100755 --- a/egs_modelscope/asr/mfcca/speech_mfcca_asr-zh-cn-16k-alimeeting-vocab4950/infer.py +++ b/egs_modelscope/asr/mfcca/speech_mfcca_asr-zh-cn-16k-alimeeting-vocab4950/infer.py @@ -18,8 +18,8 @@ def modelscope_infer_core(output_dir, split_dir, njob, idx): os.environ['CUDA_VISIBLE_DEVICES'] = str(gpu_id) inference_pipline = pipeline( task=Tasks.auto_speech_recognition, - model='yufan6/speech_mfcca_asr-zh-cn-16k-alimeeting-vocab4950', - model_revision='v2.0.0', + model='NPU-ASLP/speech_mfcca_asr-zh-cn-16k-alimeeting-vocab4950', + model_revision='v1.0.0', output_dir=output_dir_job, batch_size=1, ) diff --git a/egs_modelscope/asr/mfcca/speech_mfcca_asr-zh-cn-16k-alimeeting-vocab4950/infer_after_finetune.py b/egs_modelscope/asr/mfcca/speech_mfcca_asr-zh-cn-16k-alimeeting-vocab4950/infer_after_finetune.py index 00faad0c2..e714a3d03 100755 --- a/egs_modelscope/asr/mfcca/speech_mfcca_asr-zh-cn-16k-alimeeting-vocab4950/infer_after_finetune.py +++ b/egs_modelscope/asr/mfcca/speech_mfcca_asr-zh-cn-16k-alimeeting-vocab4950/infer_after_finetune.py @@ -59,7 +59,7 @@ def modelscope_infer_after_finetune(params): if __name__ == '__main__': params = {} - params["modelscope_model_name"] = "yufan6/speech_mfcca_asr-zh-cn-16k-alimeeting-vocab4950" + params["modelscope_model_name"] = "NPU-ASLP/speech_mfcca_asr-zh-cn-16k-alimeeting-vocab4950" params["required_files"] = ["feats_stats.npz", "decoding.yaml", "configuration.json"] params["output_dir"] = "./checkpoint" params["data_dir"] = "./example_data/validation"