From edfdc05b8c6961f7a1ba4d23f36507eb49c928c8 Mon Sep 17 00:00:00 2001
From: speech_asr <wangjiaming.wjm@alibaba-inc.com>
Date: 星期五, 10 二月 2023 09:56:39 +0800
Subject: [PATCH] update data2vec_paraformer demo
---
egs_modelscope/asr/data2vec/speech_data2vec_pretrain-paraformer-zh-cn-aishell2-16k/finetune.py | 4 ++--
1 files changed, 2 insertions(+), 2 deletions(-)
diff --git a/egs_modelscope/asr/data2vec/speech_data2vec_pretrain-paraformer-zh-cn-aishell2-16k/finetune.py b/egs_modelscope/asr/data2vec/speech_data2vec_pretrain-paraformer-zh-cn-aishell2-16k/finetune.py
index 005cae6..a5f1ee4 100644
--- a/egs_modelscope/asr/data2vec/speech_data2vec_pretrain-paraformer-zh-cn-aishell2-16k/finetune.py
+++ b/egs_modelscope/asr/data2vec/speech_data2vec_pretrain-paraformer-zh-cn-aishell2-16k/finetune.py
@@ -25,13 +25,13 @@
if __name__ == '__main__':
- params = modelscope_args(model="damo/speech_data2vec_pretrain-zh-cn-aishell2-16k-pytorch",
+ params = modelscope_args(model="damo/speech_data2vec_pretrain-paraformer-zh-cn-aishell2-16k",
data_path="./data")
params.output_dir = "./checkpoint"
params.data_path = "./example_data/"
params.dataset_type = "small"
params.batch_bins = 16000
params.max_epoch = 50
- params.lr = 0.00005
+ params.lr = 0.00002
modelscope_finetune(params)
--
Gitblit v1.9.1