From b1ff4b75b7fec62aeb7a1f29f122dc0b2ad281bd Mon Sep 17 00:00:00 2001
From: riship <riship@nvidia.com>
Date: Wed, 8 Jan 2025 16:03:19 -0800
Subject: [PATCH] updating hyperparams for knn+neighborsampling, necesarry to
 get the golden triples in our retrieved subgraph before filtering with PCST

---
 examples/llm/hotpot_qa.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/examples/llm/hotpot_qa.py b/examples/llm/hotpot_qa.py
index eb27d48c07b9..750b8f0d4e28 100644
--- a/examples/llm/hotpot_qa.py
+++ b/examples/llm/hotpot_qa.py
@@ -115,7 +115,7 @@
     # k for KNN
     knn_neighsample_bs = 4096
     # number of neighbors for each seed node selected by KNN
-    fanout = 200
+    fanout = 100
     # number of hops for neighborsampling
     num_hops = 2
     query_loader = RAGQueryLoader(