reduce layers

kumo-ai · Nov 6, 2024 · b43a444 · b43a444
1 parent 0e09e23
commit b43a444
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/benchmark/tgt_ijcai_benchmark.py b/benchmark/tgt_ijcai_benchmark.py
@@ -231,7 +231,7 @@ def create_edge(data, behavior, beh_idx, pkey_name, pkey_idx):
 elif args.model in ["contextgnn", "shallowrhsgnn"]:
     model_search_space = {
         "encoder_channels": [32, 64, 128, 256],
-        "encoder_layers": [2, 4, 8],
+        "encoder_layers": [2, 4],
         "channels": [32, 64, 128, 256],
         "embedding_dim": [32, 64, 128, 256],
         "norm": ["layer_norm", "batch_norm"],