Added TODO

GreeleyGroup · Sep 23, 2023 · 2671896 · 2671896
1 parent a9d3d7a
commit 2671896
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/src/models.py b/src/models.py
@@ -98,6 +98,7 @@ def init_conv_layers(self):
         for i in range(self.n_partitions):
             part_conv_layers = []
             for j in range(self.n_conv[i]):
+                # TODO Add possibility of changing convolutional layers
                 conv_layer = [
                     gnn.CGConv(
                         channels=self.conv_size[i],