From fa1bcb0081df2967c4eb90ef75629e20a297a168 Mon Sep 17 00:00:00 2001
From: wassname <wassname@users.noreply.github.com>
Date: Sun, 1 Nov 2020 09:50:55 +0800
Subject: [PATCH] misc

---
 seq2seq_time/models/tcn.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/seq2seq_time/models/tcn.py b/seq2seq_time/models/tcn.py
index baceed3..4ab5a45 100644
--- a/seq2seq_time/models/tcn.py
+++ b/seq2seq_time/models/tcn.py
@@ -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