misc

2026-06-27 16:31:46 +08:00 · 2020-11-01 09:50:55 +08:00
parent 1b7000d089
commit fa1bcb0081
1 changed files with 2 additions and 2 deletions
@@ -159,11 +159,11 @@ class TCNSeq2Seq(nn.Module):
        nlayers=6,
        kernel_size=2,
        dropout=0.2,
-        embedding_dim=2,
    ):
        super().__init__()
        self.tcn = TemporalConvNet(
-            num_inputs=x_dim+y_dim,
+            num_inputs=x_dim + y_dim,
+            kernel_size=kernel_size,
            num_channels=[hidden_size] * nlayers,
            dropout=dropout)
        self._min_std = 0.01