change optimizer same with released model

pull/3023/head
zxcd 3 years ago
parent 10475bc70d
commit 61c9dbe8e0

@ -139,12 +139,10 @@ n_epoch: 80
accum_grad: 1 accum_grad: 1
global_grad_clip: 5.0 global_grad_clip: 5.0
model_optim: adadelta model_optim: sgd
model_optim_conf: model_optim_conf:
lr: 1.0 lr: 1.0
weight_decay: 0.0 weight_decay: 0.0
rho: 0.95
epsilon: 1.0e-8
wav2vec2_optim: adam wav2vec2_optim: adam
wav2vec2_optim_conf: wav2vec2_optim_conf:

@ -4,9 +4,9 @@ set -e
. ./path.sh || exit 1; . ./path.sh || exit 1;
. ./cmd.sh || exit 1; . ./cmd.sh || exit 1;
gpus=0,1,2,3 gpus=0
stage=0 stage=1
stop_stage=4 stop_stage=1
conf_path=conf/wav2vec2ASR.yaml conf_path=conf/wav2vec2ASR.yaml
ips= #xx.xx.xx.xx,xx.xx.xx.xx ips= #xx.xx.xx.xx,xx.xx.xx.xx
decode_conf_path=conf/tuning/decode.yaml decode_conf_path=conf/tuning/decode.yaml
@ -19,7 +19,7 @@ audio_file=data/demo_002_en.wav
avg_ckpt=avg_${avg_num} avg_ckpt=avg_${avg_num}
ckpt=$(basename ${conf_path} | awk -F'.' '{print $1}') 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 if [ ${stage} -le 0 ] && [ ${stop_stage} -ge 0 ]; then
# prepare data # prepare data

Loading…
Cancel
Save