examples/aishell/transformer/run.sh
@@ -15,9 +15,9 @@ 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 @@ -26,6 +26,8 @@ # exp tag tag="exp1" workspace=`pwd` master_port=12345 . utils/parse_options.sh || exit 1; @@ -115,6 +117,7 @@ torchrun \ --nnodes 1 \ --nproc_per_node ${gpu_num} \ --master_port ${master_port} \ ../../../funasr/bin/train.py \ --config-path "${workspace}/conf" \ --config-name "${config}" \
