diff --git a/models/gnn.py b/models/gnn.py
index 1f2c8ff..7414009 100644
--- a/models/gnn.py
+++ b/models/gnn.py
@@ -19,8 +19,8 @@ class Smoother(tf.keras.Model):
         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.conv4 = GCNConv(64, activation='tanh', use_bias=False)
-        self.conv5 = GCNConv(64, 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)
 
     def call(self, inputs, training=None):
@@ -28,8 +28,8 @@ class Smoother(tf.keras.Model):
         x = self.conv1([x, self.a], training=training)
         x = self.conv2([x, self.a], training=training)
         x = self.conv3([x, self.a], training=training)
-        x = self.conv4([x, self.a], training=training)
-        x = self.conv5([x, self.a], training=training)
+        # x = self.conv4([x, self.a], training=training)
+        # x = self.conv5([x, self.a], training=training)
         x = self.conv6([x, self.a], training=training)
         return x