diff --git a/egs_modelscope/asr/TEMPLATE/README.md b/egs_modelscope/asr/TEMPLATE/README.md index a5d7d6ea8..a739dbe31 100644 --- a/egs_modelscope/asr/TEMPLATE/README.md +++ b/egs_modelscope/asr/TEMPLATE/README.md @@ -85,13 +85,13 @@ Undo FunASR also offer recipes [infer.sh](https://github.com/alibaba-damo-academy/FunASR/blob/main/egs_modelscope/asr/TEMPLATE/infer.sh) to decode with multi-thread CPUs, or multi GPUs. - Setting parameters in `infer.sh` - - `model`: model name on ModelScope - - `data_dir`: the dataset dir needs to include `${data_dir}/wav.scp`. If `${data_dir}/text` is also exists, CER will be computed - - `output_dir`: result dir - - `batch_size`: batchsize of inference - - `gpu_inference`: whether to perform gpu decoding, set false for cpu decoding - - `gpuid_list`: set gpus, e.g., `gpuid_list`="0,1" - - `njob`: the number of jobs for CPU decoding, if `gpu_inference`=false, use CPU decoding, please set `njob` + - `model`: model name in [model zoo](https://alibaba-damo-academy.github.io/FunASR/en/modelscope_models.html#pretrained-models-on-modelscope), or model path in local disk + - `data_dir`: the dataset dir needs to include `wav.scp`. If `${data_dir}/text` is also exists, CER will be computed + - `output_dir`: output dir of the recognition results + - `batch_size`: `64` (Default), batch size of inference on gpu + - `gpu_inference`: `true` (Default), whether to perform gpu decoding, set false for CPU inference + - `gpuid_list`: `0,1` (Default), which gpu_ids are used to infer + - `njob`: only used for CPU inference (`gpu_inference`=`false`), `64` (Default), the number of jobs for CPU decoding - `checkpoint_dir`: only used for infer finetuned models, the path dir of finetuned models - `checkpoint_name`: only used for infer finetuned models, `valid.cer_ctc.ave.pb` (Default), which checkpoint is used to infer diff --git a/egs_modelscope/asr/TEMPLATE/infer.sh b/egs_modelscope/asr/TEMPLATE/infer.sh index 0d54abc97..134a8adce 100644 --- a/egs_modelscope/asr/TEMPLATE/infer.sh +++ b/egs_modelscope/asr/TEMPLATE/infer.sh @@ -12,7 +12,7 @@ output_dir="./results" batch_size=64 gpu_inference=true # whether to perform gpu decoding gpuid_list="0,1" # set gpus, e.g., gpuid_list="0,1" -njob=4 # the number of jobs for CPU decoding, if gpu_inference=false, use CPU decoding, please set njob +njob=64 # the number of jobs for CPU decoding, if gpu_inference=false, use CPU decoding, please set njob checkpoint_dir= checkpoint_name="valid.cer_ctc.ave.pb"