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

This commit is contained in:
jingyong hou 2021-12-03 10:25:02 +08:00
parent 56bdce8775
commit 8ea11dc572

View File

@ -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,