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__() super(MDTC, self).__init__()
assert kernel_size % 2 == 1 assert kernel_size % 2 == 1
self.kernel_size = kernel_size 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.causal = causal
self.preprocessor = TCNBlock(in_channels, self.preprocessor = TCNBlock(in_channels,
res_channels, res_channels,