| | |
|---|
| | | 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 |
|---|
| | | |
|---|
| | |
|---|
| | | # exp tag |
|---|
| | | tag="exp1" |
|---|
| | | workspace=`pwd` |
|---|
| | | |
|---|
| | | master_port=12345 |
|---|
| | | |
|---|
| | | . utils/parse_options.sh || exit 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 |
|---|
| | |
|---|
| | | torchrun \ |
|---|
| | | --nnodes 1 \ |
|---|
| | | --nproc_per_node ${gpu_num} \ |
|---|
| | | --master_port ${master_port} \ |
|---|
| | | ../../../funasr/bin/train.py \ |
|---|
| | | --config-path "${workspace}/conf" \ |
|---|
| | | --config-name "${config}" \ |
|---|
