fix bug in checking kernel size, kernel size should be a odd number
This commit is contained in:
parent
56bdce8775
commit
8ea11dc572
@ -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,
|
||||
|
||||
Loading…
x
Reference in New Issue
Block a user