From 6aba0e3b5689b5b24ba9c56d9947ce1a859b2ad9 Mon Sep 17 00:00:00 2001
From: 游雁 <zhifu.gzf@alibaba-inc.com>
Date: 星期四, 22 二月 2024 12:06:40 +0800
Subject: [PATCH] train finetune demo
---
examples/industrial_data_pretraining/paraformer/finetune.sh | 6 ++++--
1 files changed, 4 insertions(+), 2 deletions(-)
diff --git a/examples/industrial_data_pretraining/paraformer/finetune_from_model_hub.sh b/examples/industrial_data_pretraining/paraformer/finetune.sh
similarity index 89%
rename from examples/industrial_data_pretraining/paraformer/finetune_from_model_hub.sh
rename to examples/industrial_data_pretraining/paraformer/finetune.sh
index 0ccb83a..21b29b6 100644
--- a/examples/industrial_data_pretraining/paraformer/finetune_from_model_hub.sh
+++ b/examples/industrial_data_pretraining/paraformer/finetune.sh
@@ -1,6 +1,8 @@
# Copyright FunASR (https://github.com/alibaba-damo-academy/FunASR). All Rights Reserved.
# MIT License (https://opensource.org/licenses/MIT)
+# method1, finetune from model hub
+
# which gpu to train or finetune
export CUDA_VISIBLE_DEVICES="0,1"
gpu_num=$(echo $CUDA_VISIBLE_DEVICES | awk -F "," '{print NF}')
@@ -30,8 +32,8 @@
--nnodes 1 \
--nproc_per_node ${gpu_num} \
funasr/bin/train.py \
-+model="damo/speech_paraformer-large_asr_nat-zh-cn-16k-common-vocab8404-pytorch" \
-+model_revision="v2.0.4" \
+++model="damo/speech_paraformer-large_asr_nat-zh-cn-16k-common-vocab8404-pytorch" \
+++model_revision="v2.0.4" \
++train_data_set_list="${train_data}" \
++valid_data_set_list="${val_data}" \
++dataset_conf.batch_size=32 \
--
Gitblit v1.9.1