From 2e36e738ca39afc8d02f3d11013bd12f937cc874 Mon Sep 17 00:00:00 2001
From: zhaomingwork <61895407+zhaomingwork@users.noreply.github.com>
Date: 星期三, 08 十一月 2023 09:22:06 +0800
Subject: [PATCH] fix bug for h5 hotwords (#1067)
---
egs/alimeeting/diarization/sond/unit_test.py | 10 +++++-----
1 files changed, 5 insertions(+), 5 deletions(-)
diff --git a/egs/alimeeting/diarization/sond/unit_test.py b/egs/alimeeting/diarization/sond/unit_test.py
index 0ecf0ee..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,7 +66,7 @@
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 = [[
"data/unit_test/raw_inputs/record.wav",
--
Gitblit v1.9.1