This commit is contained in:
02Bigboy 2022-11-07 23:15:02 +08:00
parent 75c9cbd250
commit 4de344c16d

View File

@ -217,7 +217,7 @@ class MDTC(nn.Module):
super(MDTC, self).__init__()
assert kernel_size % 2 == 1
self.kernel_size = kernel_size
assert causal == True, "we now only support causal mdtc"
assert causal is True, "we now only support causal mdtc"
self.causal = causal
self.preprocessor = TCNBlock(in_channels,
res_channels,