diff --git a/examples/aishell/s1/local/test.sh b/examples/aishell/s1/local/test.sh index 1886590f..f7e99ad7 100755 --- a/examples/aishell/s1/local/test.sh +++ b/examples/aishell/s1/local/test.sh @@ -17,7 +17,7 @@ config_path=$1 ckpt_prefix=$2 chunk_mode=false -if [[ ${config_path} =~ ^chunk_ ]];then +if [[ ${config_path} =~ ^.*chunk_.*yaml$ ]];then chunk_mode=true fi diff --git a/examples/librispeech/s1/local/test.sh b/examples/librispeech/s1/local/test.sh index fe01d700..3bd3f0bb 100755 --- a/examples/librispeech/s1/local/test.sh +++ b/examples/librispeech/s1/local/test.sh @@ -17,9 +17,10 @@ config_path=$1 ckpt_prefix=$2 chunk_mode=false -if [[ ${config_path} =~ ^chunk_ ]];then +if [[ ${config_path} =~ ^.*chunk_.*yaml$ ]];then chunk_mode=true fi +echo "chunk mode ${chunk_mode}" # download language model diff --git a/examples/timit/s1/local/test.sh b/examples/timit/s1/local/test.sh index fe01d700..a137924e 100755 --- a/examples/timit/s1/local/test.sh +++ b/examples/timit/s1/local/test.sh @@ -17,7 +17,7 @@ config_path=$1 ckpt_prefix=$2 chunk_mode=false -if [[ ${config_path} =~ ^chunk_ ]];then +if [[ ${config_path} =~ ^.*chunk_.*yaml$ ]];then chunk_mode=true fi diff --git a/examples/tiny/s1/local/test.sh b/examples/tiny/s1/local/test.sh index eb5e995d..4d3ed081 100755 --- a/examples/tiny/s1/local/test.sh +++ b/examples/tiny/s1/local/test.sh @@ -16,7 +16,7 @@ config_path=$1 ckpt_prefix=$2 chunk_mode=false -if [[ ${config_path} =~ ^chunk_ ]];then +if [[ ${config_path} =~ ^.*chunk_.*yaml$ ]];then chunk_mode=true fi