|
|
|
@ -4,7 +4,7 @@ source path.sh
|
|
|
|
|
|
|
|
|
|
gpus=0,1,2,3
|
|
|
|
|
stage=0
|
|
|
|
|
stop_stage=100
|
|
|
|
|
stop_stage=50
|
|
|
|
|
conf_path=conf/conformer.yaml
|
|
|
|
|
decode_conf_path=conf/tuning/decode.yaml
|
|
|
|
|
avg_num=20
|
|
|
|
@ -40,7 +40,7 @@ if [ ${stage} -le 4 ] && [ ${stop_stage} -ge 4 ]; then
|
|
|
|
|
CUDA_VISIBLE_DEVICES=0 ./local/align.sh ${conf_path} ${decode_conf_path} exp/${ckpt}/checkpoints/${avg_ckpt} || exit -1
|
|
|
|
|
fi
|
|
|
|
|
|
|
|
|
|
if [ ${stage} -le 5 ] && [ ${stop_stage} -ge 5 ]; then
|
|
|
|
|
if [ ${stage} -le 51 ] && [ ${stop_stage} -ge 51 ]; then
|
|
|
|
|
# export ckpt avg_n
|
|
|
|
|
CUDA_VISIBLE_DEVICES= ./local/export.sh ${conf_path} exp/${ckpt}/checkpoints/${avg_ckpt} exp/${ckpt}/checkpoints/${avg_ckpt}.jit
|
|
|
|
|
fi
|
|
|
|
|