add hyperparams

large_benchmark/conf/model/gcn2.yaml

@@ -113,3 +113,25 @@ params:
     nonreg_weight_decay: 0
     grad_norm: null
     epochs: 500
+
+  products:
+    architecture:
+      num_layers: 5
+      hidden_channels: 128
+      dropout: 0.0
+      drop_input: false
+      batch_norm: false
+      residual: false
+      shared_weights: false
+      alpha: 0.1
+      theta: 0.5
+    num_parts: 150
+    batch_size: 1
+    max_steps: 151
+    pool_size: 1
+    num_workers: 0
+    lr: 0.001
+    reg_weight_decay: 0
+    nonreg_weight_decay: 0
+    grad_norm: null
+    epochs: 240

large_benchmark/conf/model/pna.yaml

@@ -94,7 +94,7 @@ params:
       hidden_channels: 256
       aggregators: ['mean']
       scalers: ['identity', 'amplification']
-      dropout: 0.2
+      dropout: 0.3
       drop_input: false
       batch_norm: false
       residual: false
@@ -107,5 +107,5 @@ params:
     reg_weight_decay: 0
     nonreg_weight_decay: 0
     grad_norm: null
-    epochs: 350
+    epochs: 150
     runs: 1