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

Update mdtc.py

Browse Source
This commit is contained in:
xiaohou 2021-11-30 17:05:48 +08:00 committed by GitHub
parent f642c0952a
commit 8a50252c64
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
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,10 +186,10 @@ class MDTC(nn.Module):
causal: bool, causal: bool,
): ):
super(MDTC, self).__init__() super(MDTC, self).__init__()
self.kernel_size = kernel_size
if kernel_size % 2 == 0: if kernel_size % 2 == 0:
print("The kernel size of MDTC must be an odd number") print("The kernel size of MDTC must be an odd number")
exit(1) exit(1)
self.kernel_size = kernel_size
self.causal = causal self.causal = causal
self.preprocessor = TCNBlock(in_channels, self.preprocessor = TCNBlock(in_channels,
res_channels, res_channels,