fix gpu configs

datamol-io · Aug 1, 2023 · 89ba022 · 89ba022
1 parent e3498cd
commit 89ba022
Show file tree

Hide file tree

Showing 2 changed files with 20 additions and 4 deletions.
diff --git a/expts/neurips2023_configs/config_large_gcn_gpu.yaml b/expts/neurips2023_configs/config_large_gcn_gpu.yaml
@@ -1,7 +1,15 @@
 # Testing GCN on LargeMix with FP16/32 on GPU
+
+defaults:
+  - base_config: large
+  - _self_
+
 constants:
-  name: &name neurips2023_large_data_gcn_gpu
-  config_override: "expts/neurips2023_configs/config_large_gcn.yaml"
+  name: neurips2023_large_data_gcn_gpu
+
+architecture:
+  gnn:  # Set as null to avoid a post-nn network
+    layer_type: 'pyg:gcn' #pyg:gine #'pyg:gps' # pyg:gated-gcn, pyg:gine,pyg:gps
 
 accelerator:
   type: gpu

diff --git a/expts/neurips2023_configs/config_small_gcn_gpu.yaml b/expts/neurips2023_configs/config_small_gcn_gpu.yaml
@@ -1,7 +1,15 @@
 # Testing GCN on ToyMix with FP16/32 on GPU
+
+defaults:
+  - base_config: small
+  - _self_
+
 constants:
-  name: &name neurips2023_small_data_gcn_gpu
-  config_override: "expts/neurips2023_configs/config_small_gcn.yaml"
+  name: neurips2023_small_data_gcn_gpu
+
+architecture:
+  gnn:  # Set as null to avoid a post-nn network
+    layer_type: 'pyg:gcn' #pyg:gine #'pyg:gps' # pyg:gated-gcn, pyg:gine,pyg:gps
 
 accelerator:
   type: gpu  # cpu or ipu or gpu