fix

kumo-ai · Nov 6, 2024 · 582d5ee · 582d5ee
1 parent 5bc93a2
commit 582d5ee
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/benchmark/tgt_ijcai_benchmark.py b/benchmark/tgt_ijcai_benchmark.py
@@ -231,10 +231,10 @@ def create_edge(data, behavior, beh_idx, pkey_name, pkey_idx):
     model_cls = IDGNN
 elif args.model in ["contextgnn", "shallowrhsgnn"]:
     model_search_space = {
-        "encoder_channels": [32, 64, 128, 256],
+        "encoder_channels": [32, 64, 128],
         "encoder_layers": [2, 4],
-        "channels": [32, 64, 128, 256],
-        "embedding_dim": [32, 64, 128, 256],
+        "channels": [32, 64, 128],
+        "embedding_dim": [32, 64, 128],
         "norm": ["layer_norm", "batch_norm"],
         "rhs_emb_mode": [
             RHSEmbeddingMode.FUSION, RHSEmbeddingMode.FEATURE,