diff --git a/models/gnn.py b/models/gnn.py
index 7414009..0e55b7b 100644
--- a/models/gnn.py
+++ b/models/gnn.py
@@ -16,9 +16,9 @@ class Smoother(tf.keras.Model):
     def __init__(self, ):
         super().__init__()
         self.a = sp_matrix_to_sp_tensor(sio.loadmat("custom_utils/ssmpl0.mat")['L'].tocoo())
-        self.conv1 = GCNConv(64, activation='tanh', use_bias=False)
-        self.conv2 = GCNConv(64, activation='tanh', use_bias=False)
-        self.conv3 = GCNConv(64, activation='tanh', use_bias=False)
+        self.conv1 = GCNConv(3, activation='tanh', use_bias=False)
+        self.conv2 = GCNConv(3, activation='tanh', use_bias=False)
+        self.conv3 = GCNConv(3, activation='tanh', use_bias=False)
         # self.conv4 = GCNConv(64, activation='tanh', use_bias=False)
         # self.conv5 = GCNConv(64, activation='tanh', use_bias=False)
         self.conv6 = GCNConv(3, activation='tanh', use_bias=False)