diff --git a/examples/aishell/s1/local/test.sh b/examples/aishell/s1/local/test.sh index 1886590ff..f7e99ad7f 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 fe01d7004..3bd3f0bba 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 fe01d7004..a137924e2 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 eb5e995df..4d3ed081a 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