| | |
|---|
| | | #!/usr/bin/env bash |
|---|
| | | |
|---|
| | | |
|---|
| | | CUDA_VISIBLE_DEVICES="0,1" |
|---|
| | | CUDA_VISIBLE_DEVICES="0,1,2,3" |
|---|
| | | |
|---|
| | | # general configuration |
|---|
| | | feats_dir="../DATA" #feature output dictionary |
|---|
| | |
|---|
| | | nj=32 |
|---|
| | | |
|---|
| | | inference_device="cuda" #"cpu" |
|---|
| | | inference_checkpoint="model.pt.avg5" |
|---|
| | | inference_checkpoint="model.pt.avg10" |
|---|
| | | inference_scp="wav.scp" |
|---|
| | | inference_batch_size=32 |
|---|
| | | inference_batch_size=1 |
|---|
| | | |
|---|
| | | # data |
|---|
| | | raw_data=../raw_data |
|---|
| | |
|---|
| | | # exp tag |
|---|
| | | tag="exp1" |
|---|
| | | workspace=`pwd` |
|---|
| | | |
|---|
| | | master_port=12345 |
|---|
| | | |
|---|
| | | . utils/parse_options.sh || exit 1; |
|---|
| | | |
|---|
| | |
|---|
| | | torchrun \ |
|---|
| | | --nnodes 1 \ |
|---|
| | | --nproc_per_node ${gpu_num} \ |
|---|
| | | --master_port ${master_port} \ |
|---|
| | | ../../../funasr/bin/train.py \ |
|---|
| | | --config-path "${workspace}/conf" \ |
|---|
| | | --config-name "${config}" \ |
|---|
