fix hyperparams for gat and resgated

Author: aditya0by0

configs/model/gat.yml

@@ -6,10 +6,10 @@ init_args:
     in_channels: 158 # number of node/atom properties
     hidden_channels: 256
     out_channels: 512
-    num_layers: 5
+    num_layers: 4
     edge_dim: 7 # number of bond properties
     heads: 8  # the number of heads should be divisible by output channels (hidden channels if output channel not given)
     v2: False  # set True to use `torch_geometric.nn.conv.GATv2Conv` convolution layers, default is GATConv
     dropout: 0
   n_molecule_properties: 0
-  n_linear_layers: 2
+  n_linear_layers: 1

configs/model/resgated.yml

@@ -6,8 +6,8 @@ init_args:
     in_channels: 158 # number of node/atom properties
     hidden_channels: 256
     out_channels: 512
-    num_layers: 5
+    num_layers: 4
     edge_dim: 7 # number of bond properties
     dropout: 0
   n_molecule_properties: 0
-  n_linear_layers: 2
+  n_linear_layers: 1