From 61c9dbe8e0a73c89531f8d504b74691786b16d39 Mon Sep 17 00:00:00 2001 From: zxcd <228587199@qq.com> Date: Mon, 13 Mar 2023 08:14:17 +0000 Subject: [PATCH] change optimizer same with released model --- examples/aishell/asr3/conf/wav2vec2ASR.yaml | 4 +--- examples/aishell/asr3/run.sh | 8 ++++---- 2 files changed, 5 insertions(+), 7 deletions(-) diff --git a/examples/aishell/asr3/conf/wav2vec2ASR.yaml b/examples/aishell/asr3/conf/wav2vec2ASR.yaml index cdb04f8c1..e845e8622 100755 --- a/examples/aishell/asr3/conf/wav2vec2ASR.yaml +++ b/examples/aishell/asr3/conf/wav2vec2ASR.yaml @@ -139,12 +139,10 @@ n_epoch: 80 accum_grad: 1 global_grad_clip: 5.0 -model_optim: adadelta +model_optim: sgd model_optim_conf: lr: 1.0 weight_decay: 0.0 - rho: 0.95 - epsilon: 1.0e-8 wav2vec2_optim: adam wav2vec2_optim_conf: diff --git a/examples/aishell/asr3/run.sh b/examples/aishell/asr3/run.sh index 9b0a3c472..26e7e607c 100755 --- a/examples/aishell/asr3/run.sh +++ b/examples/aishell/asr3/run.sh @@ -4,9 +4,9 @@ set -e . ./path.sh || exit 1; . ./cmd.sh || exit 1; -gpus=0,1,2,3 -stage=0 -stop_stage=4 +gpus=0 +stage=1 +stop_stage=1 conf_path=conf/wav2vec2ASR.yaml ips= #xx.xx.xx.xx,xx.xx.xx.xx decode_conf_path=conf/tuning/decode.yaml @@ -19,7 +19,7 @@ audio_file=data/demo_002_en.wav avg_ckpt=avg_${avg_num} ckpt=$(basename ${conf_path} | awk -F'.' '{print $1}') -echo "checkpoint name ${ckpt}"git revert -v +echo "checkpoint name ${ckpt}" if [ ${stage} -le 0 ] && [ ${stop_stage} -ge 0 ]; then # prepare data