From 33d3d2084403fd34b79c835d2f2fe04f6cd8f738 Mon Sep 17 00:00:00 2001
From: 游雁 <zhifu.gzf@alibaba-inc.com>
Date: 星期三, 13 九月 2023 09:33:54 +0800
Subject: [PATCH] Merge branch 'main' of github.com:alibaba-damo-academy/FunASR add
---
egs/alimeeting/diarization/sond/unit_test.py | 14 +++++++-------
1 files changed, 7 insertions(+), 7 deletions(-)
diff --git a/egs/alimeeting/diarization/sond/unit_test.py b/egs/alimeeting/diarization/sond/unit_test.py
index 10fc558..0f40ab2 100644
--- a/egs/alimeeting/diarization/sond/unit_test.py
+++ b/egs/alimeeting/diarization/sond/unit_test.py
@@ -4,7 +4,7 @@
def test_fbank_cpu_infer():
diar_config_path = "config_fbank.yaml"
- diar_model_path = "sond.pth"
+ diar_model_path = "sond.pb"
output_dir = "./outputs"
data_path_and_name_and_type = [
("data/unit_test/test_feats.scp", "speech", "kaldi_ark"),
@@ -24,7 +24,7 @@
def test_fbank_gpu_infer():
diar_config_path = "config_fbank.yaml"
- diar_model_path = "sond.pth"
+ diar_model_path = "sond.pb"
output_dir = "./outputs"
data_path_and_name_and_type = [
("data/unit_test/test_feats.scp", "speech", "kaldi_ark"),
@@ -45,10 +45,10 @@
def test_wav_gpu_infer():
diar_config_path = "config.yaml"
- diar_model_path = "sond.pth"
+ diar_model_path = "sond.pb"
output_dir = "./outputs"
data_path_and_name_and_type = [
- ("data/unit_test/test_wav.scp", "speech", "kaldi_ark"),
+ ("data/unit_test/test_wav.scp", "speech", "sound"),
("data/unit_test/test_profile.scp", "profile", "kaldi_ark"),
]
pipeline = inference_launch(
@@ -66,15 +66,15 @@
def test_without_profile_gpu_infer():
diar_config_path = "config.yaml"
- diar_model_path = "sond.pth"
+ diar_model_path = "sond.pb"
output_dir = "./outputs"
- raw_inputs = [
+ raw_inputs = [[
"data/unit_test/raw_inputs/record.wav",
"data/unit_test/raw_inputs/spk1.wav",
"data/unit_test/raw_inputs/spk2.wav",
"data/unit_test/raw_inputs/spk3.wav",
"data/unit_test/raw_inputs/spk4.wav"
- ]
+ ]]
pipeline = inference_launch(
mode="sond_demo",
diar_train_config=diar_config_path,
--
Gitblit v1.9.1