| | |
|---|
| | | #!/usr/bin/env bash |
|---|
| | | |
|---|
| | | |
|---|
| | | CUDA_VISIBLE_DEVICES="0,1" |
|---|
| | | CUDA_VISIBLE_DEVICES="0,1,2,3" |
|---|
| | | |
|---|
| | | # 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 |
|---|
