From c197b3aa85ef7de4fdcfc1f706ab95205fdb8617 Mon Sep 17 00:00:00 2001
From: 游雁 <zhifu.gzf@alibaba-inc.com>
Date: 星期一, 08 五月 2023 22:06:44 +0800
Subject: [PATCH] fix vad cpu infer.sh batch=1
---
funasr/bin/vad_inference.py | 11 ++++-------
funasr/bin/vad_inference_online.py | 4 +---
egs_modelscope/vad/TEMPLATE/infer.sh | 2 +-
3 files changed, 6 insertions(+), 11 deletions(-)
diff --git a/egs_modelscope/vad/TEMPLATE/infer.sh b/egs_modelscope/vad/TEMPLATE/infer.sh
index 7dc0387..0651c98 100644
--- a/egs_modelscope/vad/TEMPLATE/infer.sh
+++ b/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
diff --git a/funasr/bin/vad_inference.py b/funasr/bin/vad_inference.py
index f9dc397..5fbd844 100644
--- a/funasr/bin/vad_inference.py
+++ b/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)
diff --git a/funasr/bin/vad_inference_online.py b/funasr/bin/vad_inference_online.py
index e1dbcf2..a363309 100644
--- a/funasr/bin/vad_inference_online.py
+++ b/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)
--
Gitblit v1.9.1