| | |
|---|
| | | count=1 |
|---|
| | | gpu_inference=true # Whether to perform gpu decoding, set false for cpu decoding |
|---|
| | | # for gpu decoding, inference_nj=ngpu*njob; for cpu decoding, inference_nj=njob |
|---|
| | | njob=8 |
|---|
| | | njob=5 |
|---|
| | | train_cmd=utils/run.pl |
|---|
| | | infer_cmd=utils/run.pl |
|---|
| | | |
|---|
| | | # general configuration |
|---|
| | | feats_dir="../DATA" #feature output dictionary, for large data |
|---|
| | | feats_dir="../DATA" #feature output dictionary |
|---|
| | | exp_dir="." |
|---|
| | | lang=zh |
|---|
| | | dumpdir=dump/fbank |
|---|
| | |
|---|
| | | data_aishell= |
|---|
| | | |
|---|
| | | # exp tag |
|---|
| | | tag="" |
|---|
| | | tag="exp1" |
|---|
| | | |
|---|
| | | . utils/parse_options.sh || exit 1; |
|---|
| | | |
|---|
| | |
|---|
| | | model_dir="baseline_$(basename "${asr_config}" .yaml)_${feats_type}_${lang}_${token_type}_${tag}" |
|---|
| | | |
|---|
| | | inference_config=conf/decode_asr_transformer.yaml |
|---|
| | | inference_asr_model=valid.acc.ave_10best.pth |
|---|
| | | inference_asr_model=valid.acc.ave_10best.pb |
|---|
| | | |
|---|
| | | # you can set gpu num for decoding here |
|---|
| | | gpuid_list=$CUDA_VISIBLE_DEVICES # set gpus for decoding, the same as training stage by default |
|---|
| | |
|---|
| | | if [ -n "${inference_config}" ]; then |
|---|
| | | _opts+="--config ${inference_config} " |
|---|
| | | fi |
|---|
| | | ${infer_cmd} --gpu "${_ngpu}" --max-jobs-run "${_nj}" JOB=1:"${_nj}" "${_logdir}"/asr_inference.JOB.log \ |
|---|
| | | ${infer_cmd} --gpu "${_ngpu}" --max-jobs-run "${_nj}" JOB=1: "${_nj}" "${_logdir}"/asr_inference.JOB.log \ |
|---|
| | | python -m funasr.bin.asr_inference_launch \ |
|---|
| | | --batch_size 1 \ |
|---|
| | | --ngpu "${_ngpu}" \ |
|---|
