Update mdtc.py
This commit is contained in:
parent
8a50252c64
commit
b9551bb716
@ -186,9 +186,7 @@ class MDTC(nn.Module):
|
||||
causal: bool,
|
||||
):
|
||||
super(MDTC, self).__init__()
|
||||
if kernel_size % 2 == 0:
|
||||
print("The kernel size of MDTC must be an odd number")
|
||||
exit(1)
|
||||
assert kernel_size % 2 == 0
|
||||
self.kernel_size = kernel_size
|
||||
self.causal = causal
|
||||
self.preprocessor = TCNBlock(in_channels,
|
||||
|
||||
Loading…
x
Reference in New Issue
Block a user