From 8ea11dc572c624de5c90bf0eeed517ffe6132287 Mon Sep 17 00:00:00 2001 From: jingyong hou Date: Fri, 3 Dec 2021 10:25:02 +0800 Subject: [PATCH] fix bug in checking kernel size, kernel size should be a odd number --- kws/model/mdtc.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/kws/model/mdtc.py b/kws/model/mdtc.py index e0c3778..736eca4 100644 --- a/kws/model/mdtc.py +++ b/kws/model/mdtc.py @@ -186,7 +186,7 @@ class MDTC(nn.Module): causal: bool, ): super(MDTC, self).__init__() - assert kernel_size % 2 == 0 + assert kernel_size % 2 == 1 self.kernel_size = kernel_size self.causal = causal self.preprocessor = TCNBlock(in_channels,