AI/wekws
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

fix bug in checking kernel size, kernel size should be a odd number

Browse Source
This commit is contained in:
jingyong hou 2021-12-03 10:25:02 +08:00
parent 56bdce8775
commit 8ea11dc572
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
kws/model/mdtc.py
View File

@ -186,7 +186,7 @@ class MDTC(nn.Module):
causal: bool, causal: bool,
): ):
super(MDTC, self).__init__() super(MDTC, self).__init__()
assert kernel_size % 2 == 0 assert kernel_size % 2 == 1
self.kernel_size = kernel_size self.kernel_size = kernel_size
self.causal = causal self.causal = causal
self.preprocessor = TCNBlock(in_channels, self.preprocessor = TCNBlock(in_channels,