diff --git a/examples/wenetspeech/asr1/local/test_wav.sh b/examples/wenetspeech/asr1/local/test_wav.sh index 13296af2e3d5c0d3b1af753b100f348e4ad2040e..90f92d80bb2526d3896f1f623fb688273f27ad23 100755 --- a/examples/wenetspeech/asr1/local/test_wav.sh +++ b/examples/wenetspeech/asr1/local/test_wav.sh @@ -29,7 +29,7 @@ for type in attention_rescoring; do output_dir=${ckpt_prefix} mkdir -p ${output_dir} python3 -u ${BIN_DIR}/test_wav.py \ - --nproc ${ngpu} \ + --ngpu ${ngpu} \ --config ${config_path} \ --result_file ${output_dir}/${type}.rsl \ --checkpoint_path ${ckpt_prefix} \ diff --git a/tools/extras/install_kaldi.sh b/tools/extras/install_kaldi.sh index 3cdcd32d4d2aeda41f3ea21aac0518d7f70b7a8e..b93e7ecf6372c9f84391289ace2c4a1b3590b45e 100755 --- a/tools/extras/install_kaldi.sh +++ b/tools/extras/install_kaldi.sh @@ -9,6 +9,7 @@ apt-get install subversion -y KALDI_GIT="--depth 1 -b master https://github.com/kaldi-asr/kaldi.git" KALDI_DIR="$PWD/kaldi" +SHARED=false if [ ! -d "$KALDI_DIR" ]; then git clone $KALDI_GIT $KALDI_DIR @@ -23,17 +24,25 @@ git pull mkdir -p "python" touch "python/.use_default_python" +# check deps ./extras/check_dependencies.sh +# make tools make -j4 +# make src pushd ../src OPENBLAS_DIR=${KALDI_DIR}/../OpenBLAS mkdir -p ${OPENBLAS_DIR}/install -./configure --shared --use-cuda=no --static-math --mathlib=OPENBLAS --openblas-root=${OPENBLAS_DIR}/install +if [ $SHARED == true ]; + ./configure --shared --use-cuda=no --static-math --mathlib=OPENBLAS --openblas-root=${OPENBLAS_DIR}/install +else + ./configure --static --use-cuda=no --static-math --mathlib=OPENBLAS --openblas-root=${OPENBLAS_DIR}/install +fi make clean -j && make depend -j && make -j4 -popd +popd # kaldi/src + -popd +popd # kaldi/tools echo "Done installing Kaldi."