examples/aishell/conformer/run.sh
@@ -15,7 +15,7 @@ nj=32 inference_device="cuda" #"cpu", "cuda:0", "cuda:1" inference_checkpoint="model.pt.avg5" inference_checkpoint="model.pt.avg10" inference_scp="wav.scp" inference_batch_size=1 @@ -26,6 +26,8 @@ # exp tag tag="exp1" workspace=`pwd` master_port=12345 . utils/parse_options.sh || exit 1; @@ -114,6 +116,7 @@ torchrun \ --nnodes 1 \ --nproc_per_node ${gpu_num} \ --master_port ${master_port} \ ../../../funasr/bin/train.py \ --config-path "${workspace}/conf" \ --config-name "${config}" \