fix vad cpu infer.sh batch=1

游雁

2023-05-08 c197b3aa85ef7de4fdcfc1f706ab95205fdb8617

fix vad cpu infer.sh batch=1

 egs_modelscope/vad/TEMPLATE/infer.sh

@@ -9,7 +9,7 @@
model="damo/speech_fsmn_vad_zh-cn-16k-common"
data_dir="./data/test"
output_dir="./results"
batch_size=64
batch_size=1
gpu_inference=true    # whether to perform gpu decoding
gpuid_list="0,1"    # set gpus, e.g., gpuid_list="0,1"
njob=64    # the number of jobs for CPU decoding, if gpu_inference=false, use CPU decoding, please set njob

 funasr/bin/vad_inference.py

@@ -274,8 +274,7 @@
    assert check_argument_types()
    if batch_size > 1:
        raise NotImplementedError("batch decoding is not implemented")
    if ngpu > 1:
        raise NotImplementedError("only single GPU decoding is supported")


    logging.basicConfig(
        level=log_level,
@@ -286,7 +285,7 @@
        device = "cuda"
    else:
        device = "cpu"

        batch_size = 1
    # 1. Set random-seed
    set_all_random_seed(seed)

@@ -376,10 +375,7 @@
        **kwargs,
):
    assert check_argument_types()
    if batch_size > 1:
        raise NotImplementedError("batch decoding is not implemented")
    if ngpu > 1:
        raise NotImplementedError("only single GPU decoding is supported")


    logging.basicConfig(
        level=log_level,
@@ -390,6 +386,7 @@
        device = "cuda"
    else:
        device = "cpu"
        batch_size = 1

    # 1. Set random-seed
    set_all_random_seed(seed)

 funasr/bin/vad_inference_online.py

@@ -156,8 +156,6 @@
    
    if batch_size > 1:
        raise NotImplementedError("batch decoding is not implemented")
    if ngpu > 1:
        raise NotImplementedError("only single GPU decoding is supported")

    logging.basicConfig(
        level=log_level,
@@ -168,7 +166,7 @@
        device = "cuda"
    else:
        device = "cpu"

        batch_size = 1
    # 1. Set random-seed
    set_all_random_seed(seed)