diff --git a/egs/aishell/data2vec_transformer_finetune/run.sh b/egs/aishell/data2vec_transformer_finetune/run.sh index e040290a3..1583b9ee8 100755 --- a/egs/aishell/data2vec_transformer_finetune/run.sh +++ b/egs/aishell/data2vec_transformer_finetune/run.sh @@ -8,7 +8,7 @@ gpu_num=2 count=1 gpu_inference=true # Whether to perform gpu decoding, set false for cpu decoding # for gpu decoding, inference_nj=ngpu*njob; for cpu decoding, inference_nj=njob -njob=1 +njob=5 train_cmd=utils/run.pl infer_cmd=utils/run.pl @@ -52,7 +52,7 @@ test_sets="dev test" asr_config=conf/train_asr_transformer_12e_6d_3072_768.yaml model_dir="baseline_$(basename "${asr_config}" .yaml)_${lang}_${token_type}_${tag}" -inference_config=conf/decode_asr_transformer_noctc_1best.yaml +inference_config=conf/decode_asr_transformer.yaml inference_asr_model=valid.acc.ave_10best.pb # you can set gpu num for decoding here