| | |
| | | nj=32 |
| | | |
| | | inference_device="cuda" #"cpu", "cuda:0", "cuda:1" |
| | | inference_checkpoint="model.pt" |
| | | inference_checkpoint="model.pt.avg10" |
| | | inference_scp="wav.scp" |
| | | inference_batch_size=1 |
| | | |
| | |
| | | --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 |
| | |
| | | ++output_dir="${inference_dir}/${JOB}" \ |
| | | ++device="${inference_device}" \ |
| | | ++ncpu=1 \ |
| | | ++disable_log=true \ |
| | | ++batch_size="${inference_batch_size}" &> ${_logdir}/log.${JOB}.txt |
| | | }& |
| | | |