From f609d6ce50375299fef10e5d35f6fca20ebb02e0 Mon Sep 17 00:00:00 2001
From: 志浩 <neo.dzh@alibaba-inc.com>
Date: 星期四, 09 三月 2023 22:01:13 +0800
Subject: [PATCH] modify unit test for speech_diarization_sond-en-us-callhome-8k-n16k4-pytorch
---
egs/callhome/diarization/sond/sond.yaml | 1 +
egs/callhome/diarization/sond/sond_fbank.yaml | 1 +
2 files changed, 2 insertions(+), 0 deletions(-)
diff --git a/egs/callhome/diarization/sond/sond.yaml b/egs/callhome/diarization/sond/sond.yaml
index 147a14a..868163f 100644
--- a/egs/callhome/diarization/sond/sond.yaml
+++ b/egs/callhome/diarization/sond/sond.yaml
@@ -2667,6 +2667,7 @@
# pass by model, equal to feature dim
# input_size: 80
pooling_type: "window_shift"
+ batchnorm_momentum: 0.01
pool_size: 20
stride: 1
tf2torch_tensor_name_prefix_torch: encoder
diff --git a/egs/callhome/diarization/sond/sond_fbank.yaml b/egs/callhome/diarization/sond/sond_fbank.yaml
index 163efd6..fc76259 100644
--- a/egs/callhome/diarization/sond/sond_fbank.yaml
+++ b/egs/callhome/diarization/sond/sond_fbank.yaml
@@ -2666,6 +2666,7 @@
encoder_conf:
# pass by model, equal to feature dim
# input_size: 80
+ batchnorm_momentum: 0.01
pooling_type: "window_shift"
pool_size: 20
stride: 1
--
Gitblit v1.9.1