| | |
|---|
| | | |
|---|
| | | # general configuration |
|---|
| | | feats_dir="../DATA" #feature output dictionary |
|---|
| | | exp_dir="." |
|---|
| | | exp_dir=`pwd` |
|---|
| | | lang=zh |
|---|
| | | token_type=char |
|---|
| | | stage=0 |
|---|
| | |
|---|
| | | nj=32 |
|---|
| | | |
|---|
| | | inference_device="cuda" #"cpu" |
|---|
| | | inference_checkpoint="model.pt" |
|---|
| | | inference_checkpoint="model.pt.avg10" |
|---|
| | | inference_scp="wav.scp" |
|---|
| | | inference_batch_size=32 |
|---|
| | | inference_batch_size=1 |
|---|
| | | |
|---|
| | | # data |
|---|
| | | raw_data=../raw_data |
|---|
| | |
|---|
| | | --config-name "${config}" \ |
|---|
| | | ++train_data_set_list="${feats_dir}/data/${train_set}/audio_datasets.jsonl" \ |
|---|
| | | ++cmvn_file="${feats_dir}/data/${train_set}/cmvn.json" \ |
|---|
| | | ++dataset_conf.num_workers=$nj |
|---|
| | | |
|---|
| | | fi |
|---|
| | | |
|---|
| | | token_list=${feats_dir}/data/${lang}_token_list/$token_type/tokens.txt |
|---|
| | |
|---|
| | | |
|---|
| | | inference_dir="${exp_dir}/exp/${model_dir}/inference-${inference_checkpoint}/${dset}" |
|---|
| | | _logdir="${inference_dir}/logdir" |
|---|
| | | echo "inference_dir: ${inference_dir}" |
|---|
| | | |
|---|
| | | mkdir -p "${_logdir}" |
|---|
| | | data_dir="${feats_dir}/data/${dset}" |
|---|
| | |
|---|
| | | ++input="${_logdir}/keys.${JOB}.scp" \ |
|---|
| | | ++output_dir="${inference_dir}/${JOB}" \ |
|---|
| | | ++device="${inference_device}" \ |
|---|
| | | ++ncpu=1 \ |
|---|
| | | ++disable_log=true \ |
|---|
| | | ++batch_size="${inference_batch_size}" &> ${_logdir}/log.${JOB}.txt |
|---|
| | | }& |
|---|
| | | |
|---|
