diff --git a/benchmark/tgt_ijcai_benchmark.py b/benchmark/tgt_ijcai_benchmark.py
index 4a8b9af..497eac2 100644
--- 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,