diff --git a/configs/iccv/config_abl_all.py b/configs/iccv/config_abl_all.py
index 4e9f18d..558c532 100644
--- a/configs/iccv/config_abl_all.py
+++ b/configs/iccv/config_abl_all.py
@@ -19,8 +19,8 @@ config['only_curvatures'] =           False
 ablation_count = [config['use_skinning'], config['remove_global_orientation'], config['use_curvature'], config['use_extra_losses'], config['use_offsets']].count(False)
 assert ablation_count == 1 or ablation_count == 0
 
-config['gpu'] = '2'
-config['name'] = 'abl_all_gnn'
+config['gpu'] = '0'
+config['name'] = 'abl_all_gnn_2'
 config['notes'] = ''
 config['tags'] = ['iccv', 'abl_all_gnn']
 
diff --git a/models/gnn.py b/models/gnn.py
index 07d65c7..71d937a 100644
--- a/models/gnn.py
+++ b/models/gnn.py
@@ -16,11 +16,11 @@ 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(16, activation='tanh', use_bias=False)
-        self.conv2 = GCNConv(16, activation='tanh', use_bias=False)
-        self.conv3 = GCNConv(16, activation='tanh', use_bias=False)
-        # self.conv4 = GCNConv(16, activation='tanh', use_bias=False)
-        # self.conv5 = GCNConv(16, activation='tanh', use_bias=False)
+        self.conv1 = GCNConv(512, activation='tanh', use_bias=False)
+        self.conv2 = GCNConv(512, activation='tanh', use_bias=False)
+        self.conv3 = GCNConv(512, activation='tanh', use_bias=False)
+        self.conv4 = GCNConv(512, 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